Merge pull request #292 from IBM/fix/custom_path

No default custom path
IBM · Dec 4, 2024 · 8ef03b1 · 8ef03b1
2 parents 3b3418a + 48958b9
commit 8ef03b1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/terratorch/cli_tools.py b/terratorch/cli_tools.py
@@ -394,10 +394,11 @@ def instantiate_classes(self) -> None:
         elif hasattr(self.config, "predict") and hasattr(self.config.predict, "custom_modules_path"):
             custom_modules_path = self.config.predict.custom_modules_path
         else:
-            default_path = Path(".") / "custom_modules"
-            custom_modules_path = os.environ.get("TERRATORCH_CUSTOM_MODULE_PATH", default_path)
+            logger.info("No custom module is being used.")
+            custom_modules_path = None 
 
-        import_custom_modules(custom_modules_path)
+        if custom_modules_path:
+            import_custom_modules(custom_modules_path)
 
 def build_lightning_cli(
     args: ArgsType = None,