fix: py package

GitDataAI · Aug 22, 2024 · fa8707f · fa8707f
1 parent b171e34
commit fa8707f
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 6 deletions.
diff --git a/crates/jzflowsdk_py/__init__.py b/crates/jzflowsdk_py/__init__.py
@@ -0,0 +1,2 @@
+from .ipc import IPCClient, IPCError
+from .loop import simple_loop
diff --git a/crates/jzflowsdk_py/loop.py b/crates/jzflowsdk_py/loop.py
@@ -1,9 +1,8 @@
 import os
 import time
 from os import path
-import joblib
 import uuid
-from ipc import IPCClient, IPCError
+from . import IPCClient, IPCError
 import logging
 
 logging.basicConfig(level=logging.INFO)

diff --git a/script/housing-prices/regression.py b/script/housing-prices/regression.py
@@ -1,16 +1,13 @@
 import argparse
 import os
-import time
 from os import path
 import numpy as np
 import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
 import joblib
 import logging
-import uuid
 from utils import plot_learning_curve
-from ipc import IPCClient, IPCError
 from jzflowsdk import simple_loop
 
 from sklearn.model_selection import ShuffleSplit

diff --git a/script/housing-prices/utils.py b/script/housing-prices/utils.py
@@ -5,7 +5,6 @@
 from sklearn.datasets import load_digits
 from sklearn.model_selection import learning_curve
 from sklearn.model_selection import ShuffleSplit
-import ipc
 
 def plot_learning_curve(estimator, title, X, y, axes=None, ylim=None, cv=None,
                         n_jobs=None, train_sizes=np.linspace(.1, 1.0, 5)):