Add init files and import changes due to restructuring

VowpalWabbit · Jun 4, 2021 · db4d806 · db4d806
1 parent 37aa0f7
commit db4d806
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/estimators/basic-usage.py b/estimators/basic-usage.py
@@ -1,9 +1,9 @@
 import argparse, os, gzip
-import cressieread
-import ips_snips
-import mle
+from contextual_bandits import cressieread
+from contextual_bandits import ips_snips
+from contextual_bandits import mle
 import ds_parse
-import cats_utils
+from contextual_bandits import cats_utils
 
 
 def compute_estimates(log_fp, cats_transformer=None):

diff --git a/estimators/test/__init__.py b/estimators/test/__init__.py
diff --git a/estimators/test/test_pi.py b/estimators/test/test_pi.py
@@ -1,9 +1,9 @@
 import os, sys
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-import pseudo_inverse
-import ips_snips
-import cats_utils
+from slates import pseudo_inverse
+from contextual_bandits import ips_snips
+from contextual_bandits import cats_utils
 
 def test_single_slot_pi_equivalent_to_ips():
     """PI should be equivalent to IPS when there is only a single slot"""