Refactor model script preprocessing method in script_model.py

NRCan · Oct 16, 2024 · 0a11de3 · 0a11de3
1 parent d09896f
commit 0a11de3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/geo_deep_learning/tools/script_model.py b/geo_deep_learning/tools/script_model.py
@@ -9,7 +9,7 @@ def __init__(self, model, data_module):
 
     def forward(self, x):
         sample = {"image": x}
-        preprocessed = self.data_module.model_script_preprocess(sample)
+        preprocessed = self.data_module._model_script_preprocess(sample)
         return self.model(preprocessed["image"])
 
 def script_model(model, data_module):