From d041dcf4e317cdc941f5612cacb97caff17c982a Mon Sep 17 00:00:00 2001 From: Matt Watson <1389937+mattdangerw@users.noreply.github.com> Date: Mon, 4 Dec 2023 10:24:34 -0800 Subject: [PATCH] Let kagglehub select latest version (#1342) This is a newly minted feature in kagglehub. We should probably add a unit test for this, but probably after our models are actually uploaded the hub. --- keras_nlp/utils/preset_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_nlp/utils/preset_utils.py b/keras_nlp/utils/preset_utils.py index f2234f615d..3b31eb244f 100644 --- a/keras_nlp/utils/preset_utils.py +++ b/keras_nlp/utils/preset_utils.py @@ -40,7 +40,7 @@ def get_file(preset, path): # Insert the kaggle framework into the handle. if len(segments) == 3: org, model, variant = segments - kaggle_handle = f"{org}/{model}/keras/{variant}/1" + kaggle_handle = f"{org}/{model}/keras/{variant}" elif len(segments) == 4: org, model, variant, version = segments kaggle_handle = f"{org}/{model}/keras/{variant}/{version}"