deepecho 0.6.1.dev0__tar.gz → 0.7.0.dev0__tar.gz

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: deepecho
-Version: 0.6.1.dev0
+Version: 0.7.0.dev0
 Summary: Create sequential synthetic data of mixed types using a GAN.
 Author-email: "DataCebo, Inc." <info@sdv.dev>
 License: BSL-1.1
@@ -19,20 +19,24 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: <3.13,>=3.8
+Requires-Python: <3.14,>=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy>=1.21.0; python_version < "3.10"
 Requires-Dist: numpy>=1.23.3; python_version >= "3.10" and python_version < "3.12"
-Requires-Dist: numpy>=1.26.0; python_version >= "3.12"
+Requires-Dist: numpy>=1.26.0; python_version >= "3.12" and python_version < "3.13"
+Requires-Dist: numpy>=2.1.0; python_version >= "3.13"
 Requires-Dist: pandas>=1.4.0; python_version < "3.11"
 Requires-Dist: pandas>=1.5.0; python_version >= "3.11" and python_version < "3.12"
-Requires-Dist: pandas>=2.1.1; python_version >= "3.12"
+Requires-Dist: pandas>=2.1.1; python_version >= "3.12" and python_version < "3.13"
+Requires-Dist: pandas>=2.2.3; python_version >= "3.13"
 Requires-Dist: torch>=1.9.0; python_version < "3.10"
 Requires-Dist: torch>=1.11.0; python_version >= "3.10" and python_version < "3.11"
 Requires-Dist: torch>=2.0.0; python_version >= "3.11" and python_version < "3.12"
-Requires-Dist: torch>=2.2.0; python_version >= "3.12"
+Requires-Dist: torch>=2.2.0; python_version >= "3.12" and python_version < "3.13"
+Requires-Dist: torch>=2.6.0; python_version >= "3.13"
 Requires-Dist: tqdm>=4.29
 Provides-Extra: test
 Requires-Dist: pytest>=3.4.2; extra == "test"

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/deepecho/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = 'DataCebo, Inc.'
 __email__ = 'info@sdv.dev'
-__version__ = '0.6.1.dev0'
+__version__ = '0.7.0.dev0'
 __path__ = __import__('pkgutil').extend_path(__path__, __name__)
 
 from deepecho.demo import load_demo

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/deepecho/models/base.py

@@ -54,12 +54,12 @@ class DeepEcho:
                 For example, a sequence might look something like::
 
                     {
-                        "context": [1],
-                        "data": [
+                        'context': [1],
+                        'data': [
                             [1, 3, 4, 5, 11, 3, 4],
-                            [2, 2, 3, 4,  5, 1, 2],
-                            [1, 3, 4, 5,  2, 3, 1]
-                        ]
+                            [2, 2, 3, 4, 5, 1, 2],
+                            [1, 3, 4, 5, 2, 3, 1],
+                        ],
                     }
 
                 The "context" attribute maps to a list of variables which
@@ -148,7 +148,7 @@ class DeepEcho:
         if segment_size is not None and not isinstance(segment_size, int):
             if sequence_index is None:
                 raise TypeError(
-                    '`segment_size` must be of type `int` if ' 'no `sequence_index` is given.'
+                    '`segment_size` must be of type `int` if no `sequence_index` is given.'
                 )
             if data[sequence_index].dtype.kind != 'M':
                 raise TypeError(

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/deepecho/models/basic_gan.py

@@ -271,7 +271,7 @@ class BasicGANModel(DeepEcho):
     def _normalize(tensor, value, properties):
         """Normalize the value between 0 and 1 and flag nans."""
         value_idx, missing_idx = properties['indices']
-        if pd.isnull(value):
+        if pd.isna(value):
             tensor[value_idx] = 0.0
             tensor[missing_idx] = 1.0
         else:
@@ -493,12 +493,12 @@ class BasicGANModel(DeepEcho):
                 For example, a sequence might look something like::
 
                     {
-                        "context": [1],
-                        "data": [
+                        'context': [1],
+                        'data': [
                             [1, 3, 4, 5, 11, 3, 4],
-                            [2, 2, 3, 4,  5, 1, 2],
-                            [1, 3, 4, 5,  2, 3, 1]
-                        ]
+                            [2, 2, 3, 4, 5, 1, 2],
+                            [1, 3, 4, 5, 2, 3, 1],
+                        ],
                     }
 
                 The "context" attribute maps to a list of variables which

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/deepecho/models/par.py

@@ -131,7 +131,7 @@ class PARModel(DeepEcho):
                     'type': t,
                     'mu': np.nanmean(x[i]),
                     'std': np.nanstd(x[i]),
-                    'nulls': pd.isnull(x[i]).any(),
+                    'nulls': pd.isna(x[i]).any(),
                     'indices': (idx, idx + 1, idx + 2),
                 }
                 idx += 3
@@ -141,7 +141,7 @@ class PARModel(DeepEcho):
                     'type': t,
                     'min': np.nanmin(x[i]),
                     'range': np.nanmax(x[i]) - np.nanmin(x[i]),
-                    'nulls': pd.isnull(x[i]).any(),
+                    'nulls': pd.isna(x[i]).any(),
                     'indices': (idx, idx + 1, idx + 2),
                 }
                 idx += 3
@@ -150,7 +150,7 @@ class PARModel(DeepEcho):
                 idx_map[i] = {'type': t, 'indices': {}}
                 idx += 1
                 for v in set(x[i]):
-                    if pd.isnull(v):
+                    if pd.isna(v):
                         v = None
 
                     idx_map[i]['indices'][v] = idx
@@ -210,30 +210,30 @@ class PARModel(DeepEcho):
 
                 elif props['type'] in ['continuous', 'timestamp']:
                     mu_idx, sigma_idx, missing_idx = props['indices']
-                    if pd.isnull(data[key][i]) or props['std'] == 0:
+                    if pd.isna(data[key][i]) or props['std'] == 0:
                         x[mu_idx] = 0.0
                     else:
                         x[mu_idx] = (data[key][i] - props['mu']) / props['std']
 
                     x[sigma_idx] = 0.0
-                    x[missing_idx] = 1.0 if pd.isnull(data[key][i]) else 0.0
+                    x[missing_idx] = 1.0 if pd.isna(data[key][i]) else 0.0
 
                 elif props['type'] in ['count']:
                     r_idx, p_idx, missing_idx = props['indices']
-                    if pd.isnull(data[key][i]) or props['range'] == 0:
+                    if pd.isna(data[key][i]) or props['range'] == 0:
                         x[r_idx] = 0.0
                     else:
                         x[r_idx] = (data[key][i] - props['min']) / props['range']
 
                     x[p_idx] = 0.0
-                    x[missing_idx] = 1.0 if pd.isnull(data[key][i]) else 0.0
+                    x[missing_idx] = 1.0 if pd.isna(data[key][i]) else 0.0
 
                 elif props['type'] in [
                     'categorical',
                     'ordinal',
                 ]:  # categorical
                     value = data[key][i]
-                    if pd.isnull(value):
+                    if pd.isna(value):
                         value = None
                     x[props['indices'][value]] = 1.0
 
@@ -258,25 +258,25 @@ class PARModel(DeepEcho):
                 mu_idx, sigma_idx, missing_idx = props['indices']
                 x[mu_idx] = (
                     0.0
-                    if (pd.isnull(context[key]) or props['std'] == 0)
+                    if (pd.isna(context[key]) or props['std'] == 0)
                     else (context[key] - props['mu']) / props['std']
                 )
                 x[sigma_idx] = 0.0
-                x[missing_idx] = 1.0 if pd.isnull(context[key]) else 0.0
+                x[missing_idx] = 1.0 if pd.isna(context[key]) else 0.0
 
             elif props['type'] in ['count']:
                 r_idx, p_idx, missing_idx = props['indices']
                 x[r_idx] = (
                     0.0
-                    if (pd.isnull(context[key]) or props['range'] == 0)
+                    if (pd.isna(context[key]) or props['range'] == 0)
                     else (context[key] - props['min']) / props['range']
                 )
                 x[p_idx] = 0.0
-                x[missing_idx] = 1.0 if pd.isnull(context[key]) else 0.0
+                x[missing_idx] = 1.0 if pd.isna(context[key]) else 0.0
 
             elif props['type'] in ['categorical', 'ordinal']:
                 value = context[key]
-                if pd.isnull(value):
+                if pd.isna(value):
                     value = None
                 x[props['indices'][value]] = 1.0
 
@@ -295,12 +295,12 @@ class PARModel(DeepEcho):
                 For example, a sequence might look something like::
 
                     {
-                        "context": [1],
-                        "data": [
+                        'context': [1],
+                        'data': [
                             [1, 3, 4, 5, 11, 3, 4],
-                            [2, 2, 3, 4,  5, 1, 2],
-                            [1, 3, 4, 5,  2, 3, 1]
-                        ]
+                            [2, 2, 3, 4, 5, 1, 2],
+                            [1, 3, 4, 5, 2, 3, 1],
+                        ],
                     }
 
                 The "context" attribute maps to a list of variables which
@@ -406,9 +406,7 @@ class PARModel(DeepEcho):
                     p_true = X_padded[: seq_len[i], i, missing_idx]
                     p_pred = missing[: seq_len[i], i]
                     log_likelihood += torch.sum(p_true * p_pred)
-                    log_likelihood += torch.sum(
-                        (1.0 - p_true) * torch.log(1.0 - torch.exp(p_pred))
-                    )
+                    log_likelihood += torch.sum((1.0 - p_true) * torch.log(1.0 - torch.exp(p_pred)))
 
             elif props['type'] in ['count']:
                 r_idx, p_idx, missing_idx = props['indices']
@@ -428,9 +426,7 @@ class PARModel(DeepEcho):
                     p_true = X_padded[: seq_len[i], i, missing_idx]
                     p_pred = missing[: seq_len[i], i]
                     log_likelihood += torch.sum(p_true * p_pred)
-                    log_likelihood += torch.sum(
-                        (1.0 - p_true) * torch.log(1.0 - torch.exp(p_pred))
-                    )
+                    log_likelihood += torch.sum((1.0 - p_true) * torch.log(1.0 - torch.exp(p_pred)))
 
             elif props['type'] in ['categorical', 'ordinal']:
                 idx = list(props['indices'].values())

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/deepecho/sequences.py

@@ -59,7 +59,7 @@ def segment_by_time(sequence, segment_size, sequence_index):
     while start <= max_time:
         end = start + segment_size
         selected = (start <= sequence_index) & (sequence_index < end)
-        sequences.append(sequence[selected.values].reset_index(drop=True))
+        sequences.append(sequence[selected.to_numpy()].reset_index(drop=True))
         start = end
 
     return sequences
@@ -112,7 +112,7 @@ def _convert_to_dicts(segments, context_columns):
             if len(context.drop_duplicates()) > 1:
                 raise ValueError('Context columns are not constant within each segment.')
 
-            context = context.iloc[0].values
+            context = context.iloc[0].to_numpy()
             segment = segment.drop(context_columns, axis=1)
         else:
             context = []
@@ -180,7 +180,7 @@ def assemble_sequences(
         segments = []
         groupby_columns = entity_columns[0] if len(entity_columns) == 1 else entity_columns
         for _, sequence in data.groupby(groupby_columns):
-            sequence.drop(entity_columns, axis=1, inplace=True)
+            sequence = sequence.drop(entity_columns, axis=1)
             if context_columns:
                 if len(sequence[context_columns].drop_duplicates()) > 1:
                     raise ValueError('Context columns are not constant within each entity.')

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/deepecho.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: deepecho
-Version: 0.6.1.dev0
+Version: 0.7.0.dev0
 Summary: Create sequential synthetic data of mixed types using a GAN.
 Author-email: "DataCebo, Inc." <info@sdv.dev>
 License: BSL-1.1
@@ -19,20 +19,24 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: <3.13,>=3.8
+Requires-Python: <3.14,>=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy>=1.21.0; python_version < "3.10"
 Requires-Dist: numpy>=1.23.3; python_version >= "3.10" and python_version < "3.12"
-Requires-Dist: numpy>=1.26.0; python_version >= "3.12"
+Requires-Dist: numpy>=1.26.0; python_version >= "3.12" and python_version < "3.13"
+Requires-Dist: numpy>=2.1.0; python_version >= "3.13"
 Requires-Dist: pandas>=1.4.0; python_version < "3.11"
 Requires-Dist: pandas>=1.5.0; python_version >= "3.11" and python_version < "3.12"
-Requires-Dist: pandas>=2.1.1; python_version >= "3.12"
+Requires-Dist: pandas>=2.1.1; python_version >= "3.12" and python_version < "3.13"
+Requires-Dist: pandas>=2.2.3; python_version >= "3.13"
 Requires-Dist: torch>=1.9.0; python_version < "3.10"
 Requires-Dist: torch>=1.11.0; python_version >= "3.10" and python_version < "3.11"
 Requires-Dist: torch>=2.0.0; python_version >= "3.11" and python_version < "3.12"
-Requires-Dist: torch>=2.2.0; python_version >= "3.12"
+Requires-Dist: torch>=2.2.0; python_version >= "3.12" and python_version < "3.13"
+Requires-Dist: torch>=2.6.0; python_version >= "3.13"
 Requires-Dist: tqdm>=4.29
 Provides-Extra: test
 Requires-Dist: pytest>=3.4.2; extra == "test"

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/deepecho.egg-info/requires.txt

@@ -17,11 +17,16 @@ numpy>=1.23.3
 pandas>=1.5.0
 torch>=2.0.0
 
-[:python_version >= "3.12"]
+[:python_version >= "3.12" and python_version < "3.13"]
 numpy>=1.26.0
 pandas>=2.1.1
 torch>=2.2.0
 
+[:python_version >= "3.13"]
+numpy>=2.1.0
+pandas>=2.2.3
+torch>=2.6.0
+
 [dev]
 deepecho[test]
 setuptools<70

{deepecho-0.6.1.dev0 → deepecho-0.7.0.dev0}/pyproject.toml

@@ -13,24 +13,28 @@ classifiers = [
         'Programming Language :: Python :: 3.10',
         'Programming Language :: Python :: 3.11',
         'Programming Language :: Python :: 3.12',
+        'Programming Language :: Python :: 3.13',
         'Topic :: Scientific/Engineering :: Artificial Intelligence',
 ]
 keywords = ['deepecho', 'DeepEcho']
 dynamic = ['version']
 license = { text = 'BSL-1.1' }
-requires-python = '>=3.8,<3.13'
+requires-python = '>=3.8,<3.14'
 readme = 'README.md'
 dependencies = [
     "numpy>=1.21.0;python_version<'3.10'",
     "numpy>=1.23.3;python_version>='3.10' and python_version<'3.12'",
-    "numpy>=1.26.0;python_version>='3.12'",
+    "numpy>=1.26.0;python_version>='3.12' and python_version<'3.13'",
+    "numpy>=2.1.0;python_version>='3.13'",
     "pandas>=1.4.0;python_version<'3.11'",
     "pandas>=1.5.0;python_version>='3.11' and python_version<'3.12'",
-    "pandas>=2.1.1;python_version>='3.12'",
+    "pandas>=2.1.1;python_version>='3.12' and python_version<'3.13'",
+    "pandas>=2.2.3;python_version>='3.13'",
     "torch>=1.9.0;python_version<'3.10'",
     "torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",
     "torch>=2.0.0;python_version>='3.11' and python_version<'3.12'",
-    "torch>=2.2.0;python_version>='3.12'",
+    "torch>=2.2.0;python_version>='3.12' and python_version<'3.13'",
+    "torch>=2.6.0;python_version>='3.13'",
     'tqdm>=4.29',
 ]
 
@@ -122,7 +126,7 @@ use_parentheses = true
 collect_ignore = ['pyproject.toml']
 
 [tool.bumpversion]
-current_version = "0.6.1.dev0"
+current_version = "0.7.0.dev0"
 parse = '(?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\.(?P<release>[a-z]+)(?P<candidate>\d+))?'
 serialize = [
     '{major}.{minor}.{patch}.{release}{candidate}',
@@ -160,7 +164,8 @@ build-backend = 'setuptools.build_meta'
 
 [tool.ruff]
 preview = true
-line-length = 99
+line-length = 100
+indent-width = 4
 src = ["deepecho"]
 target-version = "py312"
 exclude = [
@@ -168,7 +173,9 @@ exclude = [
     ".tox",
     ".git",
     "__pycache__",
-    ".ipynb_checkpoints"
+    "*.ipynb",
+    ".ipynb_checkpoints",
+    "tasks.py",
 ]
 
 [tool.ruff.lint]
@@ -178,13 +185,23 @@ select = [
     # Pycodestyle
     "E",
     "W",
+    # pydocstyle
+    "D",
     # isort
-    "I001"
+    "I001",
+    # print statements
+    "T201",
+    # pandas-vet
+    "PD",
+    # numpy 2.0
+    "NPY201"
 ]
 ignore = [
-    "E501",
+    # pydocstyle
     "D107",  # Missing docstring in __init__
     "D417",   # Missing argument descriptions in the docstring, this is a bug from pydocstyle: https://github.com/PyCQA/pydocstyle/issues/449
+    "PD901",
+    "PD101",
 ]
 
 [tool.ruff.lint.pep8-naming]
@@ -192,6 +209,7 @@ extend-ignore-names = ["X", "C", "X_padded", "Y", "Y_padded"]
 
 [tool.ruff.lint.isort]
 known-first-party = ["deepecho"]
+lines-between-types = 0
 
 [tool.ruff.lint.per-file-ignores]
 "__init__.py" = ["F401", "E402", "F403", "F405", "E501", "I001"]
@@ -200,6 +218,12 @@ known-first-party = ["deepecho"]
 quote-style = "single"
 indent-style = "space"
 preview = true
+docstring-code-format = true
+docstring-code-line-length = "dynamic"
 
 [tool.ruff.lint.pydocstyle]
 convention = "google"
+
+[tool.ruff.lint.pycodestyle]
+max-doc-length = 100
+max-line-length = 100