Fix custom_encoder imports (#12)

korbit-ai · Sep 15, 2022 · 1d9cf71 · 1d9cf71
1 parent 1081640
commit 1d9cf71
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/codejail/safe_exec.py b/codejail/safe_exec.py
@@ -144,7 +144,7 @@ def flush(self, *args, **kwargs):
 
     the_code.append(textwrap.dedent(
         """
-        from custom_encoder import GlobalEncoder
+        from codejail.custom_encoder import GlobalEncoder
         g_dict = json_safe(g_dict)
         """
         # Write the globals back to the calling process.
@@ -209,7 +209,7 @@ def decode_object(obj):
         if isinstance(obj, bytes):
             return obj.decode('utf-8')
         if type(obj).__module__ == 'numpy':
-            from custom_encoder import NumpyEncoder
+            from codejail.custom_encoder import NumpyEncoder
             return NumpyEncoder().default(obj)
         if isinstance(obj, (list, tuple)):
             new_list = []