Browse Source

Made extrapolator optional

Clemens-Alexander Brust 5 years ago
parent
commit
38dc00d9e0
1 changed files with 9 additions and 6 deletions
  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