From 7a6ef655de10a9e0bbc33f13ef91e1edd514e514 Mon Sep 17 00:00:00 2001
From: anonymousdouble <112695649+anonymousdouble@users.noreply.github.com>
Date: Tue, 19 Dec 2023 16:28:59 +1100
Subject: [PATCH] Update eval_image_retrieval.py

refactor with chain assignment to make code more Pythonic, concise and efficient.
---
 eval_image_retrieval.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/eval_image_retrieval.py b/eval_image_retrieval.py
index 999f8c900..b6bdc20b2 100644
--- a/eval_image_retrieval.py
+++ b/eval_image_retrieval.py
@@ -40,8 +40,7 @@ def __init__(self, dir_main, dataset, split, transform=None, imsize=None):
         with open(gnd_fname, 'rb') as f:
             cfg = pickle.load(f)
         cfg['gnd_fname'] = gnd_fname
-        cfg['ext'] = '.jpg'
-        cfg['qext'] = '.jpg'
+        cfg['qext'] = cfg['ext'] = '.jpg'
         cfg['dir_data'] = os.path.join(dir_main, dataset)
         cfg['dir_images'] = os.path.join(cfg['dir_data'], 'jpg')
         cfg['n'] = len(cfg['imlist'])