Safe parameter lookup

ecmwf-lab · Apr 4, 2024 · 83c6bc5 · 83c6bc5
1 parent c9c487f
commit 83c6bc5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ai_models/remote/model.py b/ai_models/remote/model.py
@@ -77,12 +77,12 @@ def load_parameters(self):
         self._param.update(params)
 
     def get_parameter(self, name):
-        if (param := self._param.get(name, None)) is not None:
+        if (param := self._param.get(name)) is not None:
             return param
 
         self._param.update(self.api.metadata(self.model, name))
 
-        return self._param[name]
+        return self._param.get(name)
 
     @cached_property
     def param_level_ml(self):