refactor

ibm-granite · Jan 28, 2025 · c4cf9c6 · c4cf9c6
1 parent be9c9aa
commit c4cf9c6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/services/inference/tsfminference/.gitignore b/services/inference/tsfminference/.gitignore
@@ -7,4 +7,3 @@ hfutil.py
 inference_payloads.py
 service_handler.py
 tsfm_config.py
-tsfm_service_handler.py
diff --git a/services/inference/tsfminference/hf_inference_handler.py b/services/inference/tsfminference/hf_inference_handler.py
@@ -10,7 +10,7 @@
 from tsfm_public.toolkit.time_series_preprocessor import extend_time_series
 from tsfm_public.toolkit.util import select_by_index
 
-from .hf_service_handler import ForecastingHuggingFaceHandler
+from .hf_service_handler import ForecastingHuggingFaceHandler, TinyTimeMixerForecastingHandler
 from .inference_payloads import ForecastingMetadataInput, ForecastingParameters
 from .service_handler import (
     ForecastingInferenceHandler,
@@ -163,3 +163,8 @@ def _calculate_data_point_counts(
         }
         LOGGER.info(f"Data point counts: {counts}")
         return counts
+
+
+class TinyTimeMixerForecastingInferenceHandler(
+    TinyTimeMixerForecastingHandler, ForecastingHuggingFaceInferenceHandler
+): ...