tirex-mirror 2025.6.9.dev3__py3-none-any.whl → 2025.8.28__py3-none-any.whl

tirex/api_adapter/forecast.py

@@ -8,20 +8,6 @@ import torch
 
 from .standard_adapter import ContextType, get_batches
 
-try:
-    from .gluon import format_gluonts_output, get_gluon_batches
-
-    _GLUONTS_AVAILABLE = True
-except ImportError:
-    _GLUONTS_AVAILABLE = False
-
-try:
-    from .hf_data import get_hfdata_batches
-
-    _HF_DATASETS_AVAILABLE = True
-except ImportError:
-    _HF_DATASETS_AVAILABLE = False
-
 
 DEF_TARGET_COLUMN = "target"
 DEF_META_COLUMNS = ("start", "item_id")
@@ -39,7 +25,9 @@ def _format_output(
     elif output_type == "numpy":
         return quantiles.cpu().numpy(), means.cpu().numpy()
     elif output_type == "gluonts":
-        if not _GLUONTS_AVAILABLE:
+        try:
+            from .gluon import format_gluonts_output
+        except ImportError:
             raise ValueError("output_type glutonts needs GluonTs but GluonTS is not available (not installed)!")
         return format_gluonts_output(quantiles, means, sample_meta, quantile_levels)
     else:
@@ -171,8 +159,11 @@ class ForecastModel(ABC):
                                           autogluon data processing function.
         """
         assert batch_size >= 1, "Batch size must be >= 1"
-        if not _GLUONTS_AVAILABLE:
+        try:
+            from .gluon import get_gluon_batches
+        except ImportError:
             raise ValueError("forecast_gluon glutonts needs GluonTs but GluonTS is not available (not installed)!")
+
         batches = get_gluon_batches(gluonDataset, batch_size, **data_kwargs)
         return _gen_forecast(
             self._forecast_quantiles, batches, output_type, quantile_levels, yield_per_batch, **predict_kwargs
@@ -199,10 +190,13 @@ class ForecastModel(ABC):
                                           datasets data processing function.
         """
         assert batch_size >= 1, "Batch size must be >= 1"
-        if not _HF_DATASETS_AVAILABLE:
+        try:
+            from .hf_data import get_hfdata_batches
+        except ImportError:
             raise ValueError(
                 "forecast_hfdata glutonts needs HuggingFace datasets but datasets is not available (not installed)!"
             )
+
         batches = get_hfdata_batches(hf_dataset, batch_size, **data_kwargs)
         return _gen_forecast(
             self._forecast_quantiles, batches, output_type, quantile_levels, yield_per_batch, **predict_kwargs

{tirex_mirror-2025.6.9.dev3.dist-info → tirex_mirror-2025.8.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tirex-mirror
-Version: 2025.6.9.dev3
+Version: 2025.8.28
 Summary: Unofficial mirror of NX-AI/tirex for packaging
 Author-email: Arpad Rozsas <rozsasarpi@gmail.com>
 License: NXAI COMMUNITY LICENSE AGREEMENT
@@ -80,10 +80,13 @@ Provides-Extra: gluonts
 Requires-Dist: gluonts; extra == "gluonts"
 Provides-Extra: hfdataset
 Requires-Dist: datasets; extra == "hfdataset"
+Provides-Extra: test
+Requires-Dist: fev; extra == "test"
 Provides-Extra: all
 Requires-Dist: ipykernel; extra == "all"
 Requires-Dist: gluonts; extra == "all"
 Requires-Dist: datasets; extra == "all"
+Requires-Dist: fev; extra == "all"
 Dynamic: license-file
 
 # tirex-mirror

{tirex_mirror-2025.6.9.dev3.dist-info → tirex_mirror-2025.8.28.dist-info}/RECORD

@@ -1,7 +1,7 @@
 tirex/__init__.py,sha256=rfsOeCJ7eRqU3K3TOhfN5-4XUuZFqt11wBRxk5SoAWA,292
 tirex/base.py,sha256=F18v9tTbLH0-nX-PC6kBAkYQHkS1T_7OQD6_aN6EjMw,2623
 tirex/api_adapter/__init__.py,sha256=YnTtPf5jGqvhfqoX8Ku7Yd0xohy0MmocE2ryrXVnQ1Q,135
-tirex/api_adapter/forecast.py,sha256=JyKw_d5v7nJbNzcckqx3Khm4TXdlkOY6ZmXaknSTWgQ,8625
+tirex/api_adapter/forecast.py,sha256=iOVP_L7fYlp1ZjyrQe2b8fwuEcxTYOszfZ5f9VDqKHU,8503
 tirex/api_adapter/gluon.py,sha256=faiYyn0kBBVQKbpWqrVoyylxZUrmr-qce66twpguVds,1827
 tirex/api_adapter/hf_data.py,sha256=T1eaxqC3OO9yOzIvw4sr55x6iA2AHKJTZd36rROM4fQ,1377
 tirex/api_adapter/standard_adapter.py,sha256=bI3XGYlWQu5EDyhDZyYqOJMbwi5h1aovPQvfHuWETJk,2618
@@ -10,10 +10,10 @@ tirex/models/components.py,sha256=sluhMbV6KL3W1ESoC5Nyoxdge9WSNx98alc8NG85dv0,49
 tirex/models/mixed_stack.py,sha256=ffpdhwCrPAbpp4_s1q8Z0Ei7iZ2TsqzVzOPe3BQPW9w,4790
 tirex/models/predict_utils.py,sha256=QUMZZ4_Sxa09UaHs1DG-MbfP8j_XwYt0x1zemdSEcFI,2749
 tirex/models/tirex.py,sha256=bFxtcpQB9-Hnayy_4bqif-o75DwO3-W0wJxelS8F_6c,9243
-tirex_mirror-2025.6.9.dev3.dist-info/licenses/LICENSE,sha256=HlwHKnGTlE2oNm6734V-Vy62zlkWohnuZpYXSdkqDk4,7362
-tirex_mirror-2025.6.9.dev3.dist-info/licenses/LICENSE_MIRROR.txt,sha256=ulPZMcOZdN7JvISjiID3KUwovTjrPwiMv5ku9dM7nls,496
-tirex_mirror-2025.6.9.dev3.dist-info/licenses/NOTICE.txt,sha256=rcgDscFHb-uuZO3L0_vIxYhTYl-a2Rm0lBpp3_kKdFQ,147
-tirex_mirror-2025.6.9.dev3.dist-info/METADATA,sha256=dxO9C6tCyFHqdNwbDb3eYbPS58fOSF7pX5MRM2F-Lx8,10941
-tirex_mirror-2025.6.9.dev3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tirex_mirror-2025.6.9.dev3.dist-info/top_level.txt,sha256=AOLDhfv0F_7nn3pFq0Kapg6Ky_28I_cGDXzQX3w9eO4,6
-tirex_mirror-2025.6.9.dev3.dist-info/RECORD,,
+tirex_mirror-2025.8.28.dist-info/licenses/LICENSE,sha256=HlwHKnGTlE2oNm6734V-Vy62zlkWohnuZpYXSdkqDk4,7362
+tirex_mirror-2025.8.28.dist-info/licenses/LICENSE_MIRROR.txt,sha256=ulPZMcOZdN7JvISjiID3KUwovTjrPwiMv5ku9dM7nls,496
+tirex_mirror-2025.8.28.dist-info/licenses/NOTICE.txt,sha256=rcgDscFHb-uuZO3L0_vIxYhTYl-a2Rm0lBpp3_kKdFQ,147
+tirex_mirror-2025.8.28.dist-info/METADATA,sha256=c9lhDLJfwEsYyxjEJ732XMn1FgCowXZ5yIyWK2NBy8o,11029
+tirex_mirror-2025.8.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tirex_mirror-2025.8.28.dist-info/top_level.txt,sha256=AOLDhfv0F_7nn3pFq0Kapg6Ky_28I_cGDXzQX3w9eO4,6
+tirex_mirror-2025.8.28.dist-info/RECORD,,