From 4b626306aa9a8bc1d1aae54b3a361bde8a724481 Mon Sep 17 00:00:00 2001 From: Rajat Sen Date: Fri, 13 Sep 2024 22:31:01 +0000 Subject: [PATCH] changes to __init__ --- src/timesfm/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/timesfm/__init__.py b/src/timesfm/__init__.py index c94e31e..f24f31d 100644 --- a/src/timesfm/__init__.py +++ b/src/timesfm/__init__.py @@ -12,11 +12,14 @@ # See the License for the specific language governing permissions and # limitations under the License. """TimesFM init file.""" - +print( + "TimesFM v1.2.0. See https://github.com/google-research/timesfm/blob/master/README.md for updated APIs." +) from timesfm.timesfm_base import freq_map, TimesFmCheckpoint, TimesFmHparams, TimesFmBase try: + print("Loaded Jax TimesFM.") from timesfm.timesfm_jax import TimesFmJax as TimesFm from timesfm import data_loader except Exception as _: - print("No pax dependencies installed.") + print("Loaded PyTorch TimesFM.") from timesfm.timesfm_torch import TimesFmTorch as TimesFm