tirex-mirror 2025.10.7__tar.gz → 2025.10.8__tar.gz

{tirex_mirror-2025.10.7 → tirex_mirror-2025.10.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tirex-mirror
-Version: 2025.10.7
+Version: 2025.10.8
 Summary: Unofficial mirror of NX-AI/tirex for packaging
 Author-email: Arpad Rozsas <rozsasarpi@gmail.com>
 License: NXAI COMMUNITY LICENSE AGREEMENT
@@ -63,37 +63,39 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 License-File: LICENSE_MIRROR.txt
 License-File: NOTICE.txt
-Requires-Dist: torch
-Requires-Dist: einops
-Requires-Dist: huggingface-hub
-Requires-Dist: numpy
+Requires-Dist: torch==2.8.0
+Requires-Dist: einops==0.8.1
+Requires-Dist: huggingface-hub==0.34.4
+Requires-Dist: numpy==2.1.2
 Provides-Extra: cuda
-Requires-Dist: xlstm; extra == "cuda"
-Requires-Dist: ninja; extra == "cuda"
+Requires-Dist: xlstm==2.0.4; extra == "cuda"
+Requires-Dist: ninja==1.13.0; extra == "cuda"
 Provides-Extra: notebooks
-Requires-Dist: ipykernel; extra == "notebooks"
-Requires-Dist: matplotlib; extra == "notebooks"
-Requires-Dist: pandas; extra == "notebooks"
-Requires-Dist: python-dotenv; extra == "notebooks"
+Requires-Dist: ipykernel==6.30.1; extra == "notebooks"
+Requires-Dist: matplotlib==3.10.6; extra == "notebooks"
+Requires-Dist: pandas==2.3.1; extra == "notebooks"
+Requires-Dist: python-dotenv==1.1.1; extra == "notebooks"
 Provides-Extra: gluonts
-Requires-Dist: gluonts; extra == "gluonts"
-Requires-Dist: pandas; extra == "gluonts"
+Requires-Dist: gluonts==0.16.2; extra == "gluonts"
+Requires-Dist: pandas==2.3.1; extra == "gluonts"
 Provides-Extra: hfdataset
-Requires-Dist: datasets; extra == "hfdataset"
+Requires-Dist: datasets==3.6.0; extra == "hfdataset"
 Provides-Extra: test
 Requires-Dist: fev>=0.6.0; extra == "test"
-Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest==8.4.2; extra == "test"
+Requires-Dist: requests==2.32.3; extra == "test"
 Provides-Extra: all
-Requires-Dist: xlstm; extra == "all"
-Requires-Dist: ninja; extra == "all"
-Requires-Dist: ipykernel; extra == "all"
-Requires-Dist: matplotlib; extra == "all"
-Requires-Dist: pandas; extra == "all"
-Requires-Dist: python-dotenv; extra == "all"
-Requires-Dist: gluonts; extra == "all"
-Requires-Dist: datasets; extra == "all"
-Requires-Dist: pytest; extra == "all"
-Requires-Dist: fev>=0.6.0; extra == "all"
+Requires-Dist: xlstm==2.0.4; extra == "all"
+Requires-Dist: ninja==1.13.0; extra == "all"
+Requires-Dist: ipykernel==6.30.1; extra == "all"
+Requires-Dist: matplotlib==3.10.6; extra == "all"
+Requires-Dist: pandas==2.3.1; extra == "all"
+Requires-Dist: python-dotenv==1.1.1; extra == "all"
+Requires-Dist: gluonts==0.16.2; extra == "all"
+Requires-Dist: datasets==3.6.0; extra == "all"
+Requires-Dist: pytest==8.4.2; extra == "all"
+Requires-Dist: fev==0.6.0; extra == "all"
+Requires-Dist: requests==2.32.3; extra == "all"
 Dynamic: license-file
 
 # tirex-mirror

{tirex_mirror-2025.10.7 → tirex_mirror-2025.10.8}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "tirex-mirror"
-version = "2025.10.07"
+version = "2025.10.08"
 description = "Unofficial mirror of NX-AI/tirex for packaging"
 readme = "README.md"
 requires-python = ">=3.11"
 classifiers = [ "Programming Language :: Python :: 3", "Operating System :: OS Independent",]
 keywords = [ "TiRex", "xLSTM", "Time Series", "Zero-shot", "Deep Learning",]
-dependencies = [ "torch", "einops", "huggingface-hub", "numpy",]
+dependencies = [ "torch==2.8.0", "einops==0.8.1", "huggingface-hub==0.34.4", "numpy==2.1.2",]
 [[project.authors]]
 name = "Arpad Rozsas"
 email = "rozsasarpi@gmail.com"
@@ -23,12 +23,12 @@ Repository = "https://github.com/rozsasarpi/tirex-mirror"
 Issues = "https://github.com/rozsasarpi/tirex-mirror/issues"
 
 [project.optional-dependencies]
-cuda = [ "xlstm", "ninja",]
-notebooks = [ "ipykernel", "matplotlib", "pandas", "python-dotenv",]
-gluonts = [ "gluonts", "pandas",]
-hfdataset = [ "datasets",]
-test = [ "fev>=0.6.0", "pytest",]
-all = [ "xlstm", "ninja", "ipykernel", "matplotlib", "pandas", "python-dotenv", "gluonts", "datasets", "pytest", "fev>=0.6.0",]
+cuda = [ "xlstm==2.0.4", "ninja==1.13.0",]
+notebooks = [ "ipykernel==6.30.1", "matplotlib==3.10.6", "pandas==2.3.1", "python-dotenv==1.1.1",]
+gluonts = [ "gluonts==0.16.2", "pandas==2.3.1",]
+hfdataset = [ "datasets==3.6.0",]
+test = [ "fev>=0.6.0", "pytest==8.4.2", "requests==2.32.3",]
+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",]
 
 [tool.docformatter]
 diff = false

{tirex_mirror-2025.10.7 → tirex_mirror-2025.10.8}/src/tirex_mirror.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tirex-mirror
-Version: 2025.10.7
+Version: 2025.10.8
 Summary: Unofficial mirror of NX-AI/tirex for packaging
 Author-email: Arpad Rozsas <rozsasarpi@gmail.com>
 License: NXAI COMMUNITY LICENSE AGREEMENT
@@ -63,37 +63,39 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 License-File: LICENSE_MIRROR.txt
 License-File: NOTICE.txt
-Requires-Dist: torch
-Requires-Dist: einops
-Requires-Dist: huggingface-hub
-Requires-Dist: numpy
+Requires-Dist: torch==2.8.0
+Requires-Dist: einops==0.8.1
+Requires-Dist: huggingface-hub==0.34.4
+Requires-Dist: numpy==2.1.2
 Provides-Extra: cuda
-Requires-Dist: xlstm; extra == "cuda"
-Requires-Dist: ninja; extra == "cuda"
+Requires-Dist: xlstm==2.0.4; extra == "cuda"
+Requires-Dist: ninja==1.13.0; extra == "cuda"
 Provides-Extra: notebooks
-Requires-Dist: ipykernel; extra == "notebooks"
-Requires-Dist: matplotlib; extra == "notebooks"
-Requires-Dist: pandas; extra == "notebooks"
-Requires-Dist: python-dotenv; extra == "notebooks"
+Requires-Dist: ipykernel==6.30.1; extra == "notebooks"
+Requires-Dist: matplotlib==3.10.6; extra == "notebooks"
+Requires-Dist: pandas==2.3.1; extra == "notebooks"
+Requires-Dist: python-dotenv==1.1.1; extra == "notebooks"
 Provides-Extra: gluonts
-Requires-Dist: gluonts; extra == "gluonts"
-Requires-Dist: pandas; extra == "gluonts"
+Requires-Dist: gluonts==0.16.2; extra == "gluonts"
+Requires-Dist: pandas==2.3.1; extra == "gluonts"
 Provides-Extra: hfdataset
-Requires-Dist: datasets; extra == "hfdataset"
+Requires-Dist: datasets==3.6.0; extra == "hfdataset"
 Provides-Extra: test
 Requires-Dist: fev>=0.6.0; extra == "test"
-Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest==8.4.2; extra == "test"
+Requires-Dist: requests==2.32.3; extra == "test"
 Provides-Extra: all
-Requires-Dist: xlstm; extra == "all"
-Requires-Dist: ninja; extra == "all"
-Requires-Dist: ipykernel; extra == "all"
-Requires-Dist: matplotlib; extra == "all"
-Requires-Dist: pandas; extra == "all"
-Requires-Dist: python-dotenv; extra == "all"
-Requires-Dist: gluonts; extra == "all"
-Requires-Dist: datasets; extra == "all"
-Requires-Dist: pytest; extra == "all"
-Requires-Dist: fev>=0.6.0; extra == "all"
+Requires-Dist: xlstm==2.0.4; extra == "all"
+Requires-Dist: ninja==1.13.0; extra == "all"
+Requires-Dist: ipykernel==6.30.1; extra == "all"
+Requires-Dist: matplotlib==3.10.6; extra == "all"
+Requires-Dist: pandas==2.3.1; extra == "all"
+Requires-Dist: python-dotenv==1.1.1; extra == "all"
+Requires-Dist: gluonts==0.16.2; extra == "all"
+Requires-Dist: datasets==3.6.0; extra == "all"
+Requires-Dist: pytest==8.4.2; extra == "all"
+Requires-Dist: fev==0.6.0; extra == "all"
+Requires-Dist: requests==2.32.3; extra == "all"
 Dynamic: license-file
 
 # tirex-mirror

{tirex_mirror-2025.10.7 → tirex_mirror-2025.10.8}/src/tirex_mirror.egg-info/SOURCES.txt

@@ -26,6 +26,7 @@ src/tirex_mirror.egg-info/top_level.txt
 tests/test_chronos_zs.py
 tests/test_forecast.py
 tests/test_forecast_adapter.py
+tests/test_jupyterlab.py
 tests/test_slstm_torch_vs_cuda.py
 tests/test_standard_adapter.py
 tests/test_util_freq.py

tirex_mirror-2025.10.8/src/tirex_mirror.egg-info/requires.txt

@@ -0,0 +1,39 @@
+torch==2.8.0
+einops==0.8.1
+huggingface-hub==0.34.4
+numpy==2.1.2
+
+[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
+
+[cuda]
+xlstm==2.0.4
+ninja==1.13.0
+
+[gluonts]
+gluonts==0.16.2
+pandas==2.3.1
+
+[hfdataset]
+datasets==3.6.0
+
+[notebooks]
+ipykernel==6.30.1
+matplotlib==3.10.6
+pandas==2.3.1
+python-dotenv==1.1.1
+
+[test]
+fev>=0.6.0
+pytest==8.4.2
+requests==2.32.3

{tirex_mirror-2025.10.7 → tirex_mirror-2025.10.8}/tests/test_forecast.py

@@ -54,9 +54,6 @@ def test_forecast_seattle_5T(tirex_model, resample_strategy, ref_mean_path, ref_
     ref_mean = load_tensor_from_txt_file(ref_mean_path).unsqueeze(0)
     ref_quantiles = load_tensor_from_pt_file(ref_quantiles_path)
 
-    # write the resampled quantiles to a file
-    torch.save(quantiles, "loop_seattle_5T_quantiles_resampled_ref.pt")
-
     # default rtol & atol for bfloat16
     torch.testing.assert_close(mean, ref_mean, rtol=1.6e-2, atol=1e-5)
     torch.testing.assert_close(quantiles, ref_quantiles, rtol=1.6e-2, atol=1e-5)

tirex_mirror-2025.10.8/tests/test_jupyterlab.py

@@ -0,0 +1,28 @@
+# Copyright (c) NXAI GmbH.
+# This software may be used and distributed according to the terms of the NXAI Community License Agreement.
+
+import requests
+import pytest
+
+cpu_url = "http://localhost:8889"
+# gpu_url = "http://localhost:8888" - will be added as soon as self-hosted gpu runner is available
+
+def test_jupyterlab_running():
+    """Check that the JupyterLab instance inside the container is reachable."""
+    try:
+        response = requests.get(cpu_url, timeout=5)  # timeout prevents hanging
+
+        print(f"✅ Connected to {cpu_url}")
+        print("Status Code:", response.status_code)
+
+        # Basic validation
+        assert response.status_code in [200, 302], f"Unexpected status code: {response.status_code}"
+
+    except requests.exceptions.ConnectionError:
+        pytest.fail(f"❌ Could not connect to {cpu_url} (connection refused or server not running)")
+
+    except requests.exceptions.Timeout:
+        pytest.fail(f"⏰ Connection to {cpu_url} timed out")
+
+    except requests.exceptions.RequestException as e:
+        pytest.fail(f"⚠️ General error connecting to {cpu_url}: {e}")

tirex_mirror-2025.10.7/src/tirex_mirror.egg-info/requires.txt

@@ -1,37 +0,0 @@
-torch
-einops
-huggingface-hub
-numpy
-
-[all]
-xlstm
-ninja
-ipykernel
-matplotlib
-pandas
-python-dotenv
-gluonts
-datasets
-pytest
-fev>=0.6.0
-
-[cuda]
-xlstm
-ninja
-
-[gluonts]
-gluonts
-pandas
-
-[hfdataset]
-datasets
-
-[notebooks]
-ipykernel
-matplotlib
-pandas
-python-dotenv
-
-[test]
-fev>=0.6.0
-pytest