Minor bug fixes for agatha ml

JSybrandt · May 25, 2020 · a03c6ff · a03c6ff
1 parent cb2c71d
commit a03c6ff
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/agatha/ml/hypothesis_predictor/hypothesis_predictor.py b/agatha/ml/hypothesis_predictor/hypothesis_predictor.py
@@ -301,8 +301,6 @@ def _step(
         print("Positive Predicates:\n", positive_predicates)
         print("Positive Scores:\n", positive_predictions)
         print("Negative Scores:\n", negative_predictions)
-        print("Positive Sample:\n", pos)
-        print("Negative Sample:\n", neg)
         raise Exception("Invalid loss")
       else:
         partial_losses.append(part_loss)

diff --git a/agatha/ml/module.py b/agatha/ml/module.py
@@ -4,6 +4,7 @@
 from pytorch_lightning import Trainer
 from argparse import ArgumentParser, Namespace
 from typing import List, Any, Tuple, Dict
+import os
 
 class AgathaModule(pl.LightningModule):
   """