Procházet zdrojové kódy

Made extrapolator optional

Clemens-Alexander Brust před 5 roky
rodič
revize
38dc00d9e0
1 změnil soubory, kde provedl 9 přidání a 6 odebrání
  1. 9 6
      chillax/experiment_selfsupervised.py

+ 9 - 6
chillax/experiment_selfsupervised.py

@@ -31,13 +31,16 @@ def main(config_files):
 
     with experiment_container.exception_shroud:
         obs.log_info("Hello!")
+
         # Now, build the extrapolator
-        extrapolator = chillax_extrapolator.CHILLAXExtrapolatorFactory.create(
-            config["extrapolator"],
-            knowledge_base=experiment_container.knowledge_base,
-            observers=experiment_container.observers,
-        )
-        experiment_container.classifier.extrapolator = extrapolator
+        if "extrapolator" in config.keys():
+            extrapolator = chillax_extrapolator.CHILLAXExtrapolatorFactory.create(
+                config["extrapolator"],
+                knowledge_base=experiment_container.knowledge_base,
+                observers=experiment_container.observers,
+            )
+            experiment_container.classifier.extrapolator = extrapolator
+
         experiment_container.runner.run()
 
     # Make sure all the data is saved