sdk-seshat-python 0.5.4__tar.gz → 0.5.6__tar.gz

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sdk-seshat-python
-Version: 0.5.4
+Version: 0.5.6
 Summary: Seshat python SDK is a library to help create ML data pipelines.
 License: Commercial - see LICENSE.txt
 Author: SeshatLabs

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sdk-seshat-python"
-version = "0.5.4"
+version = "0.5.6"
 description = "Seshat python SDK is a library to help create ML data pipelines."
 authors = ["SeshatLabs <info@seshatlabs.xyz>"]
 packages = [{ include = "seshat", from = "." }]

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/general/transformer_story/base.py

@@ -2,13 +2,10 @@ import copy
 import dataclasses
 import inspect
 import re
-import unittest
 from abc import ABC, abstractmethod
 from copy import deepcopy
 from types import ModuleType
-from typing import Callable, Dict, List, Tuple, Any, Type
-
-import pytest
+from typing import Callable, Dict, List, Tuple, Any, Type, Optional
 
 from seshat.data_class import GroupSFrame, SFrame
 from seshat.transformer import Transformer
@@ -16,6 +13,18 @@ from seshat.utils.analyze_method_call import analyze_method_calls
 from seshat.utils.validation import NumericColumnValidator, TimeStampColumnValidator
 
 
+def safe_import_testcase(module_name: str, class_name: str) -> Optional[Type]:
+    """
+    Safely import a test case class. Returns None if the test module is not available.
+    This allows the package to work without test dependencies in production.
+    """
+    try:
+        module = __import__(module_name, fromlist=[class_name])
+        return getattr(module, class_name)
+    except (ImportError, ModuleNotFoundError, AttributeError):
+        return None
+
+
 class Empty:
     """
     Because of `None` itself can be default value.
@@ -154,6 +163,8 @@ class TransformerScenario:
         keep the inputs, outputs and target instance.
         Supports both unittest.TestCase and pytest test classes.
         """
+        import unittest
+
         scenarios = []
         description = [""]  # Use list to can update it inplace.
         transformer.__call__ = cls.get_call_wrapper(
@@ -165,9 +176,9 @@ class TransformerScenario:
             return cls._from_pytest_testcase(testcase, description, scenarios)
 
     @classmethod
-    def _from_unittest_testcase(
-        cls, testcase: Type[unittest.TestCase], description, scenarios
-    ):
+    def _from_unittest_testcase(cls, testcase: Type, description, scenarios):
+        import unittest
+
         suite = unittest.TestLoader().loadTestsFromTestCase(testcase)
         for test in suite:
             # Remove `test_` from the beginning of the test name, make it readable
@@ -181,6 +192,8 @@ class TransformerScenario:
         """
         Process a pytest test case to extract transformer scenarios.
         """
+        import pytest
+
         test_module_path = inspect.getfile(testcase)
 
         test_methods = cls._find_pytest_method(testcase)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/deriver/product_sum.py

@@ -8,6 +8,7 @@ from seshat.data_class import SFrame
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.deriver.base import SFrameDeriver
 
@@ -81,8 +82,7 @@ class ProductSumDeriverStory(BaseTransformerStory):
     tags = ["deriver", "feature-engineering", "weighted-sum"]
 
     def get_scenarios(self):
-        from test import ProductSumDeriverDFTestCase
-
-        return TransformerScenario.from_testcase(
-            ProductSumDeriverDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "ProductSumDeriverDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/deriver/rank.py

@@ -5,6 +5,7 @@ from seshat.data_class import SFrame
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.deriver.base import SFrameDeriver
 
@@ -80,8 +81,7 @@ class RankDeriverStory(BaseTransformerStory):
     tags = ["deriver", "ranking", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import RankDeriverDFTestCase
-
-        return TransformerScenario.from_testcase(
-            RankDeriverDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "RankDeriverDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/deriver/sender_receiver_tokens.py

@@ -8,6 +8,7 @@ from seshat.general import configs
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.deriver.base import SFrameDeriver
 from seshat.transformer.vectorizer import CosineSimilarityVectorizer
@@ -118,11 +119,10 @@ class SenderReceiverTokensDeriverStory(BaseTransformerStory):
     tags = ["deriver", "multi-sf-operation"]
 
     def get_scenarios(self):
-        from test import SenderReceiverTokensDFTestCase
-
-        return TransformerScenario.from_testcase(
-            SenderReceiverTokensDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "SenderReceiverTokensDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)
 
     interactions = [
         (

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/deriver/shift.py

@@ -7,6 +7,7 @@ from seshat.data_class import SFrame
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.deriver.base import SFrameDeriver
 
@@ -135,8 +136,7 @@ class ShiftDeriverStory(BaseTransformerStory):
     tags = ["deriver", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import ShiftDeriverDFTestCase
-
-        return TransformerScenario.from_testcase(
-            ShiftDeriverDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "ShiftDeriverDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/deriver/static_value_column_adder.py

@@ -8,6 +8,7 @@ from seshat.general.exceptions import InvalidArgumentsError
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.deriver.base import SFrameDeriver
 
@@ -91,8 +92,7 @@ class StaticValueColumnAdderStory(BaseTransformerStory):
     tags = ["deriver", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import StaticValueColumnAdderDFTestCase
-
-        return TransformerScenario.from_testcase(
-            StaticValueColumnAdderDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "StaticValueColumnAdderDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/deriver/token_last_price.py

@@ -5,6 +5,7 @@ from seshat.general import configs
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.deriver.base import SFrameDeriver
 from seshat.utils.validation import TimeStampColumnValidator
@@ -125,8 +126,7 @@ class TokenLastPriceDeriverStory(BaseTransformerStory):
     tags = ["deriver", "multi-sf-operation"]
 
     def get_scenarios(self):
-        from test import TokenLastPriceDeriverDFTestCase
-
-        return TransformerScenario.from_testcase(
-            TokenLastPriceDeriverDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "TokenLastPriceDeriverDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/trimmer/contract.py

@@ -5,6 +5,7 @@ from seshat.general import configs
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.trimmer import SFrameTrimmer
 
@@ -96,8 +97,7 @@ class ContractTrimmerStory(BaseTransformerStory):
     tags = ["trimmer", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import ContractTrimmerDFTestCase
-
-        return TransformerScenario.from_testcase(
-            ContractTrimmerDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "ContractTrimmerDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/trimmer/duplicate.py

@@ -6,6 +6,7 @@ from pyspark.sql import DataFrame as PySparkDataFrame
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.trimmer import SFrameTrimmer
 
@@ -49,8 +50,7 @@ class DuplicateTrimmerStory(BaseTransformerStory):
     tags = ["trimmer", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import DuplicateTrimmerDFTestCase
-
-        return TransformerScenario.from_testcase(
-            DuplicateTrimmerDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "DuplicateTrimmerDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/trimmer/feature.py

@@ -5,6 +5,7 @@ from seshat.general import configs
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.trimmer import SFrameTrimmer
 
@@ -51,8 +52,7 @@ class FeatureTrimmerStory(BaseTransformerStory):
     tags = ["trimmer", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import FeatureTrimmerDFTestCase
-
-        return TransformerScenario.from_testcase(
-            FeatureTrimmerDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "FeatureTrimmerDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/trimmer/group.py

@@ -2,6 +2,7 @@ from seshat.data_class import SFrame, GroupSFrame
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.trimmer import SFrameTrimmer
 
@@ -51,8 +52,7 @@ class GroupTrimmerStory(BaseTransformerStory):
     tags = ["trimmer", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import GroupTrimmerDFTestCase
-
-        return TransformerScenario.from_testcase(
-            GroupTrimmerDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "GroupTrimmerDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/trimmer/inclusion.py

@@ -5,6 +5,7 @@ from seshat.general import configs
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.trimmer import SFrameTrimmer
 
@@ -91,8 +92,7 @@ class InclusionTrimmerStory(BaseTransformerStory):
     tags = ["trimmer", "multi-sf-operation"]
 
     def get_scenarios(self):
-        from test import InclusionTrimmerDFTestCase
-
-        return TransformerScenario.from_testcase(
-            InclusionTrimmerDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "InclusionTrimmerDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/trimmer/low_transaction.py

@@ -5,6 +5,7 @@ from seshat.general import configs
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.trimmer import SFrameTrimmer
 
@@ -124,8 +125,7 @@ class LowTransactionTrimmerStory(BaseTransformerStory):
     tags = ["trimmer", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import LowTransactionDFTestCase
-
-        return TransformerScenario.from_testcase(
-            LowTransactionDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "LowTransactionDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)

{sdk_seshat_python-0.5.4 → sdk_seshat_python-0.5.6}/seshat/transformer/trimmer/zero_address.py

@@ -5,6 +5,7 @@ from seshat.general import configs
 from seshat.general.transformer_story.base import (
     BaseTransformerStory,
     TransformerScenario,
+    safe_import_testcase,
 )
 from seshat.transformer.trimmer import SFrameTrimmer
 
@@ -58,8 +59,7 @@ class ZeroAddressTrimmerStory(BaseTransformerStory):
     tags = ["trimmer", "single-sf-operation"]
 
     def get_scenarios(self):
-        from test import ZeroAddressTrimmerDFTestCase
-
-        return TransformerScenario.from_testcase(
-            ZeroAddressTrimmerDFTestCase, transformer=self.transformer
-        )
+        testcase = safe_import_testcase("test", "ZeroAddressTrimmerDFTestCase")
+        if testcase is None:
+            return []
+        return TransformerScenario.from_testcase(testcase, transformer=self.transformer)