FIX: avoid crash if rf model has to be obtained from S3

MeteoSwiss · Nov 14, 2024 · 9bdd03a · 9bdd03a
1 parent 7178359
commit 9bdd03a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rainforest/ml/rfdefinitions.py b/rainforest/ml/rfdefinitions.py
@@ -240,9 +240,9 @@ def read_rf(rf_name, filepath=''):
         raise IOError('RF model {:s} does not exist!'.format(rf_name))
 
     if is_compressed:
-        with gzip.open(open(ObjStorage.check_file(rf_name), 'rb')) as f: 
+        with gzip.open(open(rf_name, 'rb')) as f: 
             return MyCustomUnpickler(f).load()
     else:
-        with open(ObjStorage.check_file(rf_name), 'rb') as f: 
+        with open(rf_name, 'rb') as f: 
             return MyCustomUnpickler(f).load()