Merge pull request #58 from causy-dev/backporting-310

start backporting to 310
causy-dev · Aug 7, 2024 · 5d547f9 · 5d547f9
2 parents 06c1916 + 845e5eb
commit 5d547f9
Show file tree

Hide file tree

Showing 8 changed files with 96 additions and 62 deletions.
diff --git a/.github/workflows/python-tests.yml b/.github/workflows/python-tests.yml
@@ -14,7 +14,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.11", "3.12"]
+        python-version: ["3.10", "3.11", "3.12"]
         platform: [macos-latest, windows-latest, ubuntu-latest]
     runs-on: ${{ matrix.platform }}
     steps:

diff --git a/causy/causal_discovery/constraint/orientation_rules/pc.py b/causy/causal_discovery/constraint/orientation_rules/pc.py
@@ -62,7 +62,7 @@ def generate_restores(unapplied_actions):
     return results
 
 
-class ColliderTestConflictResolutionStrategies(enum.StrEnum):
+class ColliderTestConflictResolutionStrategies(str, enum.Enum):
     """
     Enum for the conflict resolution strategies for the ColliderTest.
     """

diff --git a/causy/contrib/graph_ui.py b/causy/contrib/graph_ui.py
@@ -17,7 +17,7 @@ class EdgeUIConfig(BaseModel):
     label: Optional[str] = None
 
 
-class ConditionalEdgeUIConfigComparison(enum.StrEnum):
+class ConditionalEdgeUIConfigComparison(str, enum.Enum):
     EQUAL = "EQUAL"
     GREATER = "GREATER"
     LESS = "LESS"

diff --git a/causy/data_loader.py b/causy/data_loader.py
@@ -10,7 +10,7 @@
 from causy.graph_utils import hash_dictionary
 
 
-class DataLoaderType(enum.StrEnum):
+class DataLoaderType(str, enum.Enum):
     DYNAMIC = "dynamic"  # python function which yields data
     JSON = "json"
     JSONL = "jsonl"

diff --git a/causy/edge_types.py b/causy/edge_types.py
@@ -14,7 +14,7 @@
 )
 
 
-class EdgeTypeEnum(enum.StrEnum):
+class EdgeTypeEnum(str, enum.Enum):
     DIRECTED = "directed"
     UNDIRECTED = "undirected"
     BIDIRECTED = "bidirected"

diff --git a/causy/models.py b/causy/models.py
@@ -32,7 +32,7 @@ def name(self) -> str:
         return serialize_module_name(self)
 
 
-class TestResultAction(enum.StrEnum):
+class TestResultAction(str, enum.Enum):
     """
     Actions that can be taken on the graph. These actions are used to keep track of the history of the graph.
     """
@@ -59,7 +59,7 @@ class TestResult(TestResultInterface):
     data: Optional[Dict] = None
 
 
-class AlgorithmReferenceType(enum.StrEnum):
+class AlgorithmReferenceType(str, enum.Enum):
     FILE = "file"
     NAME = "name"
     PYTHON_MODULE = "python_module"

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,7 +8,7 @@ license = "MIT"
 include = ["pyproject.toml", "README.md", "LICENSE", "causy", "causy/static/*", "causy/static/assets/*"]
 
 [tool.poetry.dependencies]
-python = "^3.11"
+python = "^3.10"
 scipy = "^1.14"
 typer = {extras = ["all"], version = "^0.12.3"}
 torch = "^2.4"