set_env

README.md

@@ -148,5 +148,5 @@ See the code in `tests/test_holoclean.py` for a documented usage of HoloClean.
 In order to run the test script, run the following:
 ```bash
 $ cd tests
-$ python test_holoclean.py
+$ ./start_test.sh
 ```

set_env.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+
+export HOLOCLEANHOME="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+echo "Holoclean home directory: $HOLOCLEANHOME"
+export PYTHONPATH="$PYTHONPATH:$HOLOCLEANHOME"
+export PATH="$PATH:$HOLOCLEANHOME"
+echo $PATH
+echo "Environment variables set!"

tests/start_test.sh

@@ -1,5 +1,5 @@
 # Set & move to home directory
-# source ../set_env.sh
+source ../set_env.sh
 
 # Launch test.
 echo "Launching test..."

tests/test_holoclean.py

@@ -1,7 +1,3 @@
-import sys
-
-sys.path.insert(0, '..')
-
 import holoclean
 
 from detect import NullDetector, ViolationDetector
@@ -14,18 +10,6 @@
 from repair.featurize import LangModelFeat
 
 
-def get_tid(row):
-    return row['tid'] - 1
-
-
-def get_attr(row):
-    return row['attribute'].lower()
-
-
-def get_value(row):
-    return row['correct_val'].lower()
-
-
 # 1. Setup a HoloClean session.
 hc = holoclean.HoloClean(pruning_topk=0.1, epochs=30, weight_decay=0.01, threads=20, batch_size=1, verbose=True, timeout=3*60000).session
 
@@ -44,4 +28,13 @@ def get_value(row):
 hc.repair_errors(featurizers)
 
 # 5. Evaluate the correctness of the results.
+def get_tid(row):
+    return row['tid'] - 1
+
+def get_attr(row):
+    return row['attribute'].lower()
+
+def get_value(row):
+    return row['correct_val'].lower()
+
 hc.evaluate('data', 'hospital_clean.csv', get_tid, get_attr, get_value)