upgini 1.1.316a1__tar.gz → 1.1.316a3__tar.gz

{upgini-1.1.316a1 → upgini-1.1.316a3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.316a1
+Version: 1.1.316a3
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/
@@ -31,7 +31,7 @@ Requires-Dist: levenshtein>=0.25.1
 Requires-Dist: lightgbm>=3.3.2
 Requires-Dist: numpy<=1.26.4,>=1.19.0
 Requires-Dist: pandas<3.0.0,>=1.1.0
-Requires-Dist: pydantic>=2.7.0
+Requires-Dist: pydantic<3.0.0,>1.0.0
 Requires-Dist: pyjwt>=2.8.0
 Requires-Dist: python-bidi==0.4.2
 Requires-Dist: python-dateutil>=2.8.0

{upgini-1.1.316a1 → upgini-1.1.316a3}/pyproject.toml

@@ -41,7 +41,7 @@ dependencies = [
     "lightgbm>=3.3.2",
     "numpy>=1.19.0,<=1.26.4",
     "pandas>=1.1.0,<3.0.0",
-    "pydantic>=2.7.0",
+    "pydantic>1.0.0,<3.0.0",
     "pyjwt>=2.8.0",
     "python-dateutil>=2.8.0",
     "python-json-logger>=2.0.2",
@@ -79,7 +79,7 @@ python = "3.10"
 cov = 'pytest --cov-report=term-missing --cov-config=pyproject.toml --cov=upgini --cov=tests'
 format = "black {args}"
 lint = "ruff check {args}"
-test_binary = 'pytest -s -vv tests/test_binary_dataset.py'
+test_all = 'pytest -s -vv tests'
 
 [[tool.hatch.envs.test.matrix]]
 python = ["3.8"]
@@ -103,7 +103,7 @@ dependencies = [
 #  "pytest-timeout",
   "requests-mock",
   "pytest-datafiles",
-  "pandas~={matrix:pandas}.0",
+  "pandas~={matrix:pandas}",
 ]
 
 [tool.black]
@@ -115,4 +115,5 @@ profile = "black"
 [tool.pytest.ini_options]
 pythonpath = [
   "./src"
-]
+]
+addopts="-n 4"

upgini-1.1.316a3/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.316a3"

{upgini-1.1.316a1 → upgini-1.1.316a3}/src/upgini/autofe/date.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, List, Optional, Union
 import numpy as np
 import pandas as pd
 from pandas.core.arrays.timedeltas import TimedeltaArray
-from pydantic import BaseModel, field_validator
+from pydantic import BaseModel, validator
 
 from upgini.autofe.operand import PandasOperand
 
@@ -246,7 +246,7 @@ class DatePercentile(DatePercentileBase):
         )
         return res
 
-    @field_validator("zero_bounds", mode="before")
+    @validator("zero_bounds", pre="true")
     def validate_bounds(cls, value):
         if value is None or isinstance(value, list):
             return value

{upgini-1.1.316a1 → upgini-1.1.316a3}/src/upgini/http.py

@@ -39,7 +39,6 @@ from upgini.metadata import (
 from upgini.resource_bundle import bundle
 from upgini.utils.track_info import get_track_metrics
 
-
 UPGINI_URL: str = "UPGINI_URL"
 UPGINI_API_KEY: str = "UPGINI_API_KEY"
 DEMO_API_KEY: str = "Aa4BPwGFbn1zNEXIkZ-NbhsRk0ricN6puKuga1-O5lM"
@@ -460,7 +459,11 @@ class _RestClient:
                         dumps(track_metrics).encode(),
                         "application/json",
                     ),
-                    "metrics": ("metrics.json", metrics.model_dump_json(exclude_none=True).encode(), "application/json"),
+                    "metrics": (
+                        "metrics.json",
+                        metrics.model_dump_json(exclude_none=True).encode(),
+                        "application/json",
+                    ),
                     "file": (metadata_with_md5.name, file, "application/octet-stream"),
                 }
                 if search_customization is not None:
@@ -544,7 +547,11 @@ class _RestClient:
                         dumps(get_track_metrics(self.client_ip, self.client_visitorid)).encode(),
                         "application/json",
                     ),
-                    "metrics": ("metrics.json", metrics.model_dump_json(exclude_none=True).encode(), "application/json"),
+                    "metrics": (
+                        "metrics.json",
+                        metrics.model_dump_json(exclude_none=True).encode(),
+                        "application/json",
+                    ),
                     "file": (metadata_with_md5.name, file, "application/octet-stream"),
                 }
                 if search_customization is not None:
@@ -640,7 +647,11 @@ class _RestClient:
             with open(file_path, "rb") as file:
                 files = {
                     "file": (metadata.name, file, "application/octet-stream"),
-                    "metadata": ("metadata.json", metadata.model_dump_json(exclude_none=True).encode(), "application/json"),
+                    "metadata": (
+                        "metadata.json",
+                        metadata.model_dump_json(exclude_none=True).encode(),
+                        "application/json",
+                    ),
                 }
 
                 return self._send_post_file_req_v2(api_path, files)

upgini-1.1.316a1/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.316a1"