[TASK] - PLAT-2934 - Upgrades python to 3.11.9 and changes python dep handling

Co-authored-by: Matt Kornfield <kornfield@gretel.ai>
Co-authored-by: Ryan Sadler <ryan.sadler@gretel.ai>
Co-authored-by: Taylor Mutch <taylor.mutch@gretel.ai>
GitOrigin-RevId: bdaeebc8016b06f70b476d5c9d461af4260ee1f1

Author: 4 people

requirements/base.txt

@@ -4,4 +4,4 @@ pandas>=1.1.0,<2
 protobuf>=4,<=4.24.0
 rdt>=1.2,<1.3
 sdv>=0.17,<0.18
-tqdm<5.0
+tqdm>=4.66,<5.0

src/gretel_synthetics/actgan/structures.py

@@ -5,7 +5,7 @@
 from __future__ import annotations
 
 from dataclasses import asdict, dataclass
-from enum import Enum
+from enum import StrEnum
 from typing import Any, Dict, List, TYPE_CHECKING
 
 if TYPE_CHECKING:
@@ -16,12 +16,12 @@
     from gretel_synthetics.actgan.column_encodings import ColumnEncoding
 
 
-class ColumnType(str, Enum):
+class ColumnType(StrEnum):
     CONTINUOUS = "continuous"
     DISCRETE = "discrete"
 
 
-class ConditionalVectorType(str, Enum):
+class ConditionalVectorType(StrEnum):
     SINGLE_DISCRETE = "single_discrete"
     ANYWAY = "anyway"
     # TODO: add ANYWAY_WITH_MASK type so the generator input includes a 0/1 mask

src/gretel_synthetics/timeseries_dgan/config.py

@@ -1,5 +1,5 @@
 from dataclasses import asdict, dataclass
-from enum import Enum
+from enum import Enum, StrEnum
 
 
 class OutputType(Enum):
@@ -142,7 +142,7 @@ def to_dict(self):
         return asdict(self)
 
 
-class DfStyle(str, Enum):
+class DfStyle(StrEnum):
     """Supported styles for parsing pandas DataFrames.
 
     See `train_dataframe` method in dgan.py for details.

src/gretel_synthetics/timeseries_dgan/dgan.py

@@ -1231,7 +1231,7 @@ def state_dict(self) -> Dict:
         return state
 
     @abc.abstractmethod
-    def _state_dict() -> Dict:
+    def _state_dict(self) -> Dict:
         """Subclass specific dictionary for saving and loading."""
         ...