chillax_extrapolator: Fixed wrong shortest_path in DepthSteps

chillax/chillax_extrapolator.py

@@ -172,7 +172,7 @@ class DepthStepsCHILLAXExtrapolator(CHILLAXExtrapolator):
 
             root = list(nx.topological_sort(self.rgraph))[0]
             self.uid_to_depth = {
-                concept.uid: nx.shortest_path(root, concept.uid)
+                concept.uid: nx.shortest_path(self.rgraph, root, concept.uid)
                 for concept in self.knowledge_base.concepts()
             }
 

chillax/experiment_selfsupervised.py

@@ -7,11 +7,6 @@ import config as pcfg
 import argparse
 
 
-class CheapObserver(instrumentation.Observer):
-    def update(self, message: instrumentation.Message):
-        print(f"Message: {message}")
-
-
 def main(config_files):
     # This shouldn't be necessary, but...
     helpers.setup_environment()