Update OpusMT.py

Fix #19
UKPLab · Mar 10, 2021 · 27bef67 · 27bef67
1 parent db9a17d
commit 27bef67
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/easynmt/models/OpusMT.py b/easynmt/models/OpusMT.py
@@ -33,10 +33,10 @@ def load_model(self, model_name):
             if len(self.models) >= self.max_loaded_models:
                 oldest_time = time.time()
                 oldest_model = None
-                for model_name in self.models:
-                    if self.models[model_name]['last_loaded'] <= oldest_time:
-                        oldest_model = model_name
-                        oldest_time = self.models[model_name]['last_loaded']
+                for loaded_model_name in self.models:
+                    if self.models[loaded_model_name]['last_loaded'] <= oldest_time:
+                        oldest_model = loaded_model_name
+                        oldest_time = self.models[loaded_model_name]['last_loaded']
                 del self.models[oldest_model]
 
             self.models[model_name] = {'tokenizer': tokenizer, 'model': model, 'last_loaded': time.time()}