tirex-mirror 2025.10.8__tar.gz → 2025.10.16__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {tirex_mirror-2025.10.8/src/tirex_mirror.egg-info → tirex_mirror-2025.10.16}/PKG-INFO +25 -27
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/pyproject.toml +8 -8
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16/src/tirex_mirror.egg-info}/PKG-INFO +25 -27
- tirex_mirror-2025.10.16/src/tirex_mirror.egg-info/requires.txt +37 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_jupyterlab.py +31 -7
- tirex_mirror-2025.10.8/src/tirex_mirror.egg-info/requires.txt +0 -39
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/LICENSE +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/LICENSE_MIRROR.txt +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/MANIFEST.in +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/NOTICE.txt +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/README.md +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/setup.cfg +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/__init__.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/__init__.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/forecast.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/gluon.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/hf_data.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/standard_adapter.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/base.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/__init__.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/patcher.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/slstm/block.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/slstm/cell.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/slstm/layer.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/tirex.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/util.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex_mirror.egg-info/SOURCES.txt +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex_mirror.egg-info/dependency_links.txt +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex_mirror.egg-info/top_level.txt +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_chronos_zs.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_forecast.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_forecast_adapter.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_slstm_torch_vs_cuda.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_standard_adapter.py +0 -0
- {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_util_freq.py +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: tirex-mirror
|
|
3
|
-
Version: 2025.10.
|
|
3
|
+
Version: 2025.10.16
|
|
4
4
|
Summary: Unofficial mirror of NX-AI/tirex for packaging
|
|
5
5
|
Author-email: Arpad Rozsas <rozsasarpi@gmail.com>
|
|
6
6
|
License: NXAI COMMUNITY LICENSE AGREEMENT
|
|
@@ -63,39 +63,37 @@ Description-Content-Type: text/markdown
|
|
|
63
63
|
License-File: LICENSE
|
|
64
64
|
License-File: LICENSE_MIRROR.txt
|
|
65
65
|
License-File: NOTICE.txt
|
|
66
|
-
Requires-Dist: torch
|
|
67
|
-
Requires-Dist: einops
|
|
68
|
-
Requires-Dist: huggingface-hub
|
|
69
|
-
Requires-Dist: numpy
|
|
66
|
+
Requires-Dist: torch
|
|
67
|
+
Requires-Dist: einops
|
|
68
|
+
Requires-Dist: huggingface-hub
|
|
69
|
+
Requires-Dist: numpy
|
|
70
70
|
Provides-Extra: cuda
|
|
71
|
-
Requires-Dist: xlstm
|
|
72
|
-
Requires-Dist: ninja
|
|
71
|
+
Requires-Dist: xlstm; extra == "cuda"
|
|
72
|
+
Requires-Dist: ninja; extra == "cuda"
|
|
73
73
|
Provides-Extra: notebooks
|
|
74
|
-
Requires-Dist: ipykernel
|
|
75
|
-
Requires-Dist: matplotlib
|
|
76
|
-
Requires-Dist: pandas
|
|
77
|
-
Requires-Dist: python-dotenv
|
|
74
|
+
Requires-Dist: ipykernel; extra == "notebooks"
|
|
75
|
+
Requires-Dist: matplotlib; extra == "notebooks"
|
|
76
|
+
Requires-Dist: pandas; extra == "notebooks"
|
|
77
|
+
Requires-Dist: python-dotenv; extra == "notebooks"
|
|
78
78
|
Provides-Extra: gluonts
|
|
79
|
-
Requires-Dist: gluonts
|
|
80
|
-
Requires-Dist: pandas
|
|
79
|
+
Requires-Dist: gluonts; extra == "gluonts"
|
|
80
|
+
Requires-Dist: pandas; extra == "gluonts"
|
|
81
81
|
Provides-Extra: hfdataset
|
|
82
|
-
Requires-Dist: datasets
|
|
82
|
+
Requires-Dist: datasets; extra == "hfdataset"
|
|
83
83
|
Provides-Extra: test
|
|
84
84
|
Requires-Dist: fev>=0.6.0; extra == "test"
|
|
85
|
-
Requires-Dist: pytest
|
|
86
|
-
Requires-Dist: requests==2.32.3; extra == "test"
|
|
85
|
+
Requires-Dist: pytest; extra == "test"
|
|
87
86
|
Provides-Extra: all
|
|
88
|
-
Requires-Dist: xlstm
|
|
89
|
-
Requires-Dist: ninja
|
|
90
|
-
Requires-Dist: ipykernel
|
|
91
|
-
Requires-Dist: matplotlib
|
|
92
|
-
Requires-Dist: pandas
|
|
93
|
-
Requires-Dist: python-dotenv
|
|
94
|
-
Requires-Dist: gluonts
|
|
95
|
-
Requires-Dist: datasets
|
|
96
|
-
Requires-Dist: pytest
|
|
97
|
-
Requires-Dist: fev
|
|
98
|
-
Requires-Dist: requests==2.32.3; extra == "all"
|
|
87
|
+
Requires-Dist: xlstm; extra == "all"
|
|
88
|
+
Requires-Dist: ninja; extra == "all"
|
|
89
|
+
Requires-Dist: ipykernel; extra == "all"
|
|
90
|
+
Requires-Dist: matplotlib; extra == "all"
|
|
91
|
+
Requires-Dist: pandas; extra == "all"
|
|
92
|
+
Requires-Dist: python-dotenv; extra == "all"
|
|
93
|
+
Requires-Dist: gluonts; extra == "all"
|
|
94
|
+
Requires-Dist: datasets; extra == "all"
|
|
95
|
+
Requires-Dist: pytest; extra == "all"
|
|
96
|
+
Requires-Dist: fev>=0.6.0; extra == "all"
|
|
99
97
|
Dynamic: license-file
|
|
100
98
|
|
|
101
99
|
# tirex-mirror
|
|
@@ -1,12 +1,12 @@
|
|
|
1
1
|
[project]
|
|
2
2
|
name = "tirex-mirror"
|
|
3
|
-
version = "2025.10.
|
|
3
|
+
version = "2025.10.16"
|
|
4
4
|
description = "Unofficial mirror of NX-AI/tirex for packaging"
|
|
5
5
|
readme = "README.md"
|
|
6
6
|
requires-python = ">=3.11"
|
|
7
7
|
classifiers = [ "Programming Language :: Python :: 3", "Operating System :: OS Independent",]
|
|
8
8
|
keywords = [ "TiRex", "xLSTM", "Time Series", "Zero-shot", "Deep Learning",]
|
|
9
|
-
dependencies = [ "torch
|
|
9
|
+
dependencies = [ "torch", "einops", "huggingface-hub", "numpy",]
|
|
10
10
|
[[project.authors]]
|
|
11
11
|
name = "Arpad Rozsas"
|
|
12
12
|
email = "rozsasarpi@gmail.com"
|
|
@@ -23,12 +23,12 @@ Repository = "https://github.com/rozsasarpi/tirex-mirror"
|
|
|
23
23
|
Issues = "https://github.com/rozsasarpi/tirex-mirror/issues"
|
|
24
24
|
|
|
25
25
|
[project.optional-dependencies]
|
|
26
|
-
cuda = [ "xlstm
|
|
27
|
-
notebooks = [ "ipykernel
|
|
28
|
-
gluonts = [ "gluonts
|
|
29
|
-
hfdataset = [ "datasets
|
|
30
|
-
test = [ "fev>=0.6.0", "pytest
|
|
31
|
-
all = [ "xlstm
|
|
26
|
+
cuda = [ "xlstm", "ninja",]
|
|
27
|
+
notebooks = [ "ipykernel", "matplotlib", "pandas", "python-dotenv",]
|
|
28
|
+
gluonts = [ "gluonts", "pandas",]
|
|
29
|
+
hfdataset = [ "datasets",]
|
|
30
|
+
test = [ "fev>=0.6.0", "pytest",]
|
|
31
|
+
all = [ "xlstm", "ninja", "ipykernel", "matplotlib", "pandas", "python-dotenv", "gluonts", "datasets", "pytest", "fev>=0.6.0",]
|
|
32
32
|
|
|
33
33
|
[tool.docformatter]
|
|
34
34
|
diff = false
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: tirex-mirror
|
|
3
|
-
Version: 2025.10.
|
|
3
|
+
Version: 2025.10.16
|
|
4
4
|
Summary: Unofficial mirror of NX-AI/tirex for packaging
|
|
5
5
|
Author-email: Arpad Rozsas <rozsasarpi@gmail.com>
|
|
6
6
|
License: NXAI COMMUNITY LICENSE AGREEMENT
|
|
@@ -63,39 +63,37 @@ Description-Content-Type: text/markdown
|
|
|
63
63
|
License-File: LICENSE
|
|
64
64
|
License-File: LICENSE_MIRROR.txt
|
|
65
65
|
License-File: NOTICE.txt
|
|
66
|
-
Requires-Dist: torch
|
|
67
|
-
Requires-Dist: einops
|
|
68
|
-
Requires-Dist: huggingface-hub
|
|
69
|
-
Requires-Dist: numpy
|
|
66
|
+
Requires-Dist: torch
|
|
67
|
+
Requires-Dist: einops
|
|
68
|
+
Requires-Dist: huggingface-hub
|
|
69
|
+
Requires-Dist: numpy
|
|
70
70
|
Provides-Extra: cuda
|
|
71
|
-
Requires-Dist: xlstm
|
|
72
|
-
Requires-Dist: ninja
|
|
71
|
+
Requires-Dist: xlstm; extra == "cuda"
|
|
72
|
+
Requires-Dist: ninja; extra == "cuda"
|
|
73
73
|
Provides-Extra: notebooks
|
|
74
|
-
Requires-Dist: ipykernel
|
|
75
|
-
Requires-Dist: matplotlib
|
|
76
|
-
Requires-Dist: pandas
|
|
77
|
-
Requires-Dist: python-dotenv
|
|
74
|
+
Requires-Dist: ipykernel; extra == "notebooks"
|
|
75
|
+
Requires-Dist: matplotlib; extra == "notebooks"
|
|
76
|
+
Requires-Dist: pandas; extra == "notebooks"
|
|
77
|
+
Requires-Dist: python-dotenv; extra == "notebooks"
|
|
78
78
|
Provides-Extra: gluonts
|
|
79
|
-
Requires-Dist: gluonts
|
|
80
|
-
Requires-Dist: pandas
|
|
79
|
+
Requires-Dist: gluonts; extra == "gluonts"
|
|
80
|
+
Requires-Dist: pandas; extra == "gluonts"
|
|
81
81
|
Provides-Extra: hfdataset
|
|
82
|
-
Requires-Dist: datasets
|
|
82
|
+
Requires-Dist: datasets; extra == "hfdataset"
|
|
83
83
|
Provides-Extra: test
|
|
84
84
|
Requires-Dist: fev>=0.6.0; extra == "test"
|
|
85
|
-
Requires-Dist: pytest
|
|
86
|
-
Requires-Dist: requests==2.32.3; extra == "test"
|
|
85
|
+
Requires-Dist: pytest; extra == "test"
|
|
87
86
|
Provides-Extra: all
|
|
88
|
-
Requires-Dist: xlstm
|
|
89
|
-
Requires-Dist: ninja
|
|
90
|
-
Requires-Dist: ipykernel
|
|
91
|
-
Requires-Dist: matplotlib
|
|
92
|
-
Requires-Dist: pandas
|
|
93
|
-
Requires-Dist: python-dotenv
|
|
94
|
-
Requires-Dist: gluonts
|
|
95
|
-
Requires-Dist: datasets
|
|
96
|
-
Requires-Dist: pytest
|
|
97
|
-
Requires-Dist: fev
|
|
98
|
-
Requires-Dist: requests==2.32.3; extra == "all"
|
|
87
|
+
Requires-Dist: xlstm; extra == "all"
|
|
88
|
+
Requires-Dist: ninja; extra == "all"
|
|
89
|
+
Requires-Dist: ipykernel; extra == "all"
|
|
90
|
+
Requires-Dist: matplotlib; extra == "all"
|
|
91
|
+
Requires-Dist: pandas; extra == "all"
|
|
92
|
+
Requires-Dist: python-dotenv; extra == "all"
|
|
93
|
+
Requires-Dist: gluonts; extra == "all"
|
|
94
|
+
Requires-Dist: datasets; extra == "all"
|
|
95
|
+
Requires-Dist: pytest; extra == "all"
|
|
96
|
+
Requires-Dist: fev>=0.6.0; extra == "all"
|
|
99
97
|
Dynamic: license-file
|
|
100
98
|
|
|
101
99
|
# tirex-mirror
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
torch
|
|
2
|
+
einops
|
|
3
|
+
huggingface-hub
|
|
4
|
+
numpy
|
|
5
|
+
|
|
6
|
+
[all]
|
|
7
|
+
xlstm
|
|
8
|
+
ninja
|
|
9
|
+
ipykernel
|
|
10
|
+
matplotlib
|
|
11
|
+
pandas
|
|
12
|
+
python-dotenv
|
|
13
|
+
gluonts
|
|
14
|
+
datasets
|
|
15
|
+
pytest
|
|
16
|
+
fev>=0.6.0
|
|
17
|
+
|
|
18
|
+
[cuda]
|
|
19
|
+
xlstm
|
|
20
|
+
ninja
|
|
21
|
+
|
|
22
|
+
[gluonts]
|
|
23
|
+
gluonts
|
|
24
|
+
pandas
|
|
25
|
+
|
|
26
|
+
[hfdataset]
|
|
27
|
+
datasets
|
|
28
|
+
|
|
29
|
+
[notebooks]
|
|
30
|
+
ipykernel
|
|
31
|
+
matplotlib
|
|
32
|
+
pandas
|
|
33
|
+
python-dotenv
|
|
34
|
+
|
|
35
|
+
[test]
|
|
36
|
+
fev>=0.6.0
|
|
37
|
+
pytest
|
|
@@ -1,21 +1,45 @@
|
|
|
1
1
|
# Copyright (c) NXAI GmbH.
|
|
2
2
|
# This software may be used and distributed according to the terms of the NXAI Community License Agreement.
|
|
3
3
|
|
|
4
|
-
import
|
|
4
|
+
import logging
|
|
5
|
+
import subprocess
|
|
6
|
+
|
|
5
7
|
import pytest
|
|
8
|
+
import requests
|
|
6
9
|
|
|
7
10
|
cpu_url = "http://localhost:8889"
|
|
8
11
|
# gpu_url = "http://localhost:8888" - will be added as soon as self-hosted gpu runner is available
|
|
9
12
|
|
|
13
|
+
logger = logging.getLogger(__name__)
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
def _docker_container_running() -> bool:
|
|
17
|
+
"""Return True if any Docker container is currently running."""
|
|
18
|
+
try:
|
|
19
|
+
result = subprocess.run(
|
|
20
|
+
["docker", "ps", "-q"],
|
|
21
|
+
check=True,
|
|
22
|
+
text=True,
|
|
23
|
+
stdout=subprocess.PIPE,
|
|
24
|
+
stderr=subprocess.DEVNULL,
|
|
25
|
+
)
|
|
26
|
+
except (FileNotFoundError, subprocess.CalledProcessError):
|
|
27
|
+
return False
|
|
28
|
+
|
|
29
|
+
return bool(result.stdout.strip())
|
|
30
|
+
|
|
31
|
+
|
|
32
|
+
pytestmark = pytest.mark.skipif(
|
|
33
|
+
not _docker_container_running(),
|
|
34
|
+
reason="requires Docker with a running container exposing JupyterLab",
|
|
35
|
+
)
|
|
36
|
+
|
|
37
|
+
|
|
10
38
|
def test_jupyterlab_running():
|
|
11
39
|
"""Check that the JupyterLab instance inside the container is reachable."""
|
|
12
40
|
try:
|
|
13
41
|
response = requests.get(cpu_url, timeout=5) # timeout prevents hanging
|
|
14
|
-
|
|
15
|
-
print(f"✅ Connected to {cpu_url}")
|
|
16
|
-
print("Status Code:", response.status_code)
|
|
17
|
-
|
|
18
|
-
# Basic validation
|
|
42
|
+
logger.info("✅ Connected to %s, status code: %s", cpu_url, response.status_code)
|
|
19
43
|
assert response.status_code in [200, 302], f"Unexpected status code: {response.status_code}"
|
|
20
44
|
|
|
21
45
|
except requests.exceptions.ConnectionError:
|
|
@@ -25,4 +49,4 @@ def test_jupyterlab_running():
|
|
|
25
49
|
pytest.fail(f"⏰ Connection to {cpu_url} timed out")
|
|
26
50
|
|
|
27
51
|
except requests.exceptions.RequestException as e:
|
|
28
|
-
pytest.fail(f"⚠️ General error connecting to {cpu_url}: {e}")
|
|
52
|
+
pytest.fail(f"⚠️ General error connecting to {cpu_url}: {e}")
|
|
@@ -1,39 +0,0 @@
|
|
|
1
|
-
torch==2.8.0
|
|
2
|
-
einops==0.8.1
|
|
3
|
-
huggingface-hub==0.34.4
|
|
4
|
-
numpy==2.1.2
|
|
5
|
-
|
|
6
|
-
[all]
|
|
7
|
-
xlstm==2.0.4
|
|
8
|
-
ninja==1.13.0
|
|
9
|
-
ipykernel==6.30.1
|
|
10
|
-
matplotlib==3.10.6
|
|
11
|
-
pandas==2.3.1
|
|
12
|
-
python-dotenv==1.1.1
|
|
13
|
-
gluonts==0.16.2
|
|
14
|
-
datasets==3.6.0
|
|
15
|
-
pytest==8.4.2
|
|
16
|
-
fev==0.6.0
|
|
17
|
-
requests==2.32.3
|
|
18
|
-
|
|
19
|
-
[cuda]
|
|
20
|
-
xlstm==2.0.4
|
|
21
|
-
ninja==1.13.0
|
|
22
|
-
|
|
23
|
-
[gluonts]
|
|
24
|
-
gluonts==0.16.2
|
|
25
|
-
pandas==2.3.1
|
|
26
|
-
|
|
27
|
-
[hfdataset]
|
|
28
|
-
datasets==3.6.0
|
|
29
|
-
|
|
30
|
-
[notebooks]
|
|
31
|
-
ipykernel==6.30.1
|
|
32
|
-
matplotlib==3.10.6
|
|
33
|
-
pandas==2.3.1
|
|
34
|
-
python-dotenv==1.1.1
|
|
35
|
-
|
|
36
|
-
[test]
|
|
37
|
-
fev>=0.6.0
|
|
38
|
-
pytest==8.4.2
|
|
39
|
-
requests==2.32.3
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/standard_adapter.py
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex_mirror.egg-info/dependency_links.txt
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|