Constrain some packages in gretel-synthetics, eliminate a deprecated …

…NumPy alias, and port over README GitOrigin-RevId: f510307c11b2c51e7268fb77e69a008d3e6a926c
gretelai · May 9, 2023 · 997be9e · 997be9e
1 parent 4844ead
commit 997be9e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 category-encoders==2.2.2
 loky==2.9.0
-numpy>=1.18.0
-pandas>=1.1.0
+numpy>=1.18.0,<1.24
+pandas>=1.1.0,<2
 protobuf~=3.20
 sentencepiece==0.1.97
 smart_open>=2.1.0,<6.0

diff --git a/src/gretel_synthetics/utils/stats.py b/src/gretel_synthetics/utils/stats.py
@@ -123,7 +123,7 @@ def get_numeric_distribution_bins(training: pd.Series, synthetic: pd.Series):
     else:
         min_value = min(np.nanmin(training), np.nanmin(synthetic))
         max_value = max(np.nanmax(training), np.nanmax(synthetic))
-    bins = np.array([], dtype=np.float)
+    bins = np.array([], dtype=np.float64)
 
     # Use 'doane' to find bins. 'fd' causes too many OOM issues.
     # We also bin across the training and synthetic Series combined since we are binning across the combined range, otherwise we can see OOM's or sigkill's.