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.
Files changed (35) hide show
  1. {tirex_mirror-2025.10.8/src/tirex_mirror.egg-info → tirex_mirror-2025.10.16}/PKG-INFO +25 -27
  2. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/pyproject.toml +8 -8
  3. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16/src/tirex_mirror.egg-info}/PKG-INFO +25 -27
  4. tirex_mirror-2025.10.16/src/tirex_mirror.egg-info/requires.txt +37 -0
  5. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_jupyterlab.py +31 -7
  6. tirex_mirror-2025.10.8/src/tirex_mirror.egg-info/requires.txt +0 -39
  7. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/LICENSE +0 -0
  8. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/LICENSE_MIRROR.txt +0 -0
  9. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/MANIFEST.in +0 -0
  10. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/NOTICE.txt +0 -0
  11. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/README.md +0 -0
  12. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/setup.cfg +0 -0
  13. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/__init__.py +0 -0
  14. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/__init__.py +0 -0
  15. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/forecast.py +0 -0
  16. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/gluon.py +0 -0
  17. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/hf_data.py +0 -0
  18. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/api_adapter/standard_adapter.py +0 -0
  19. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/base.py +0 -0
  20. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/__init__.py +0 -0
  21. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/patcher.py +0 -0
  22. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/slstm/block.py +0 -0
  23. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/slstm/cell.py +0 -0
  24. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/slstm/layer.py +0 -0
  25. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/models/tirex.py +0 -0
  26. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex/util.py +0 -0
  27. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex_mirror.egg-info/SOURCES.txt +0 -0
  28. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex_mirror.egg-info/dependency_links.txt +0 -0
  29. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/src/tirex_mirror.egg-info/top_level.txt +0 -0
  30. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_chronos_zs.py +0 -0
  31. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_forecast.py +0 -0
  32. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_forecast_adapter.py +0 -0
  33. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_slstm_torch_vs_cuda.py +0 -0
  34. {tirex_mirror-2025.10.8 → tirex_mirror-2025.10.16}/tests/test_standard_adapter.py +0 -0
  35. {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.8
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==2.8.0
67
- Requires-Dist: einops==0.8.1
68
- Requires-Dist: huggingface-hub==0.34.4
69
- Requires-Dist: numpy==2.1.2
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==2.0.4; extra == "cuda"
72
- Requires-Dist: ninja==1.13.0; extra == "cuda"
71
+ Requires-Dist: xlstm; extra == "cuda"
72
+ Requires-Dist: ninja; extra == "cuda"
73
73
  Provides-Extra: notebooks
74
- Requires-Dist: ipykernel==6.30.1; extra == "notebooks"
75
- Requires-Dist: matplotlib==3.10.6; extra == "notebooks"
76
- Requires-Dist: pandas==2.3.1; extra == "notebooks"
77
- Requires-Dist: python-dotenv==1.1.1; extra == "notebooks"
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==0.16.2; extra == "gluonts"
80
- Requires-Dist: pandas==2.3.1; extra == "gluonts"
79
+ Requires-Dist: gluonts; extra == "gluonts"
80
+ Requires-Dist: pandas; extra == "gluonts"
81
81
  Provides-Extra: hfdataset
82
- Requires-Dist: datasets==3.6.0; extra == "hfdataset"
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==8.4.2; extra == "test"
86
- Requires-Dist: requests==2.32.3; extra == "test"
85
+ Requires-Dist: pytest; extra == "test"
87
86
  Provides-Extra: all
88
- Requires-Dist: xlstm==2.0.4; extra == "all"
89
- Requires-Dist: ninja==1.13.0; extra == "all"
90
- Requires-Dist: ipykernel==6.30.1; extra == "all"
91
- Requires-Dist: matplotlib==3.10.6; extra == "all"
92
- Requires-Dist: pandas==2.3.1; extra == "all"
93
- Requires-Dist: python-dotenv==1.1.1; extra == "all"
94
- Requires-Dist: gluonts==0.16.2; extra == "all"
95
- Requires-Dist: datasets==3.6.0; extra == "all"
96
- Requires-Dist: pytest==8.4.2; extra == "all"
97
- Requires-Dist: fev==0.6.0; extra == "all"
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.08"
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==2.8.0", "einops==0.8.1", "huggingface-hub==0.34.4", "numpy==2.1.2",]
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==2.0.4", "ninja==1.13.0",]
27
- notebooks = [ "ipykernel==6.30.1", "matplotlib==3.10.6", "pandas==2.3.1", "python-dotenv==1.1.1",]
28
- gluonts = [ "gluonts==0.16.2", "pandas==2.3.1",]
29
- hfdataset = [ "datasets==3.6.0",]
30
- test = [ "fev>=0.6.0", "pytest==8.4.2", "requests==2.32.3",]
31
- all = [ "xlstm==2.0.4", "ninja==1.13.0", "ipykernel==6.30.1", "matplotlib==3.10.6", "pandas==2.3.1", "python-dotenv==1.1.1", "gluonts==0.16.2", "datasets==3.6.0", "pytest==8.4.2", "fev==0.6.0", "requests==2.32.3",]
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.8
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==2.8.0
67
- Requires-Dist: einops==0.8.1
68
- Requires-Dist: huggingface-hub==0.34.4
69
- Requires-Dist: numpy==2.1.2
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==2.0.4; extra == "cuda"
72
- Requires-Dist: ninja==1.13.0; extra == "cuda"
71
+ Requires-Dist: xlstm; extra == "cuda"
72
+ Requires-Dist: ninja; extra == "cuda"
73
73
  Provides-Extra: notebooks
74
- Requires-Dist: ipykernel==6.30.1; extra == "notebooks"
75
- Requires-Dist: matplotlib==3.10.6; extra == "notebooks"
76
- Requires-Dist: pandas==2.3.1; extra == "notebooks"
77
- Requires-Dist: python-dotenv==1.1.1; extra == "notebooks"
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==0.16.2; extra == "gluonts"
80
- Requires-Dist: pandas==2.3.1; extra == "gluonts"
79
+ Requires-Dist: gluonts; extra == "gluonts"
80
+ Requires-Dist: pandas; extra == "gluonts"
81
81
  Provides-Extra: hfdataset
82
- Requires-Dist: datasets==3.6.0; extra == "hfdataset"
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==8.4.2; extra == "test"
86
- Requires-Dist: requests==2.32.3; extra == "test"
85
+ Requires-Dist: pytest; extra == "test"
87
86
  Provides-Extra: all
88
- Requires-Dist: xlstm==2.0.4; extra == "all"
89
- Requires-Dist: ninja==1.13.0; extra == "all"
90
- Requires-Dist: ipykernel==6.30.1; extra == "all"
91
- Requires-Dist: matplotlib==3.10.6; extra == "all"
92
- Requires-Dist: pandas==2.3.1; extra == "all"
93
- Requires-Dist: python-dotenv==1.1.1; extra == "all"
94
- Requires-Dist: gluonts==0.16.2; extra == "all"
95
- Requires-Dist: datasets==3.6.0; extra == "all"
96
- Requires-Dist: pytest==8.4.2; extra == "all"
97
- Requires-Dist: fev==0.6.0; extra == "all"
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 requests
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