From db4d8060a178fd77fc394e9b3469640ca1239438 Mon Sep 17 00:00:00 2001
From: JuiP <juipradhan2k@gmail.com>
Date: Mon, 31 May 2021 22:43:59 +0530
Subject: [PATCH] Add init files and import changes due to restructuring

---
 estimators/basic-usage.py   | 8 ++++----
 estimators/test/__init__.py | 0
 estimators/test/test_pi.py  | 6 +++---
 3 files changed, 7 insertions(+), 7 deletions(-)
 create mode 100644 estimators/test/__init__.py

diff --git a/estimators/basic-usage.py b/estimators/basic-usage.py
index 51221ac..41dace9 100644
--- 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
new file mode 100644
index 0000000..e69de29
diff --git a/estimators/test/test_pi.py b/estimators/test/test_pi.py
index f9d3838..1fe09e5 100644
--- 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"""