RDS-193: Support extracting attributes and features from long style d…

…ata frames. GitOrigin-RevId: f6d615c29893d368c5ba13730054e58af3557afa
gretelai · Jun 24, 2022 · 2c008c0 · 2c008c0
1 parent 57c4cec
commit 2c008c0
Show file tree

Hide file tree

Showing 3 changed files with 1,162 additions and 173 deletions.
diff --git a/src/gretel_synthetics/timeseries_dgan/config.py b/src/gretel_synthetics/timeseries_dgan/config.py
@@ -136,3 +136,13 @@ def to_dict(self):
             object, e.g., `DGANConfig(**config.to_dict())`
         """
         return asdict(self)
+
+
+class DfStyle(str, Enum):
+    """Supported styles for parsing pandas DataFrames.
+
+    See `train_dataframe` method in dgan.py for details.
+    """
+
+    WIDE = "wide"
+    LONG = "long"