Fix test

Author: lucyhe

patches/kaggle_module_resolver.py

@@ -11,12 +11,12 @@ def _is_on_kaggle_notebook():
     return os.getenv("KAGGLE_KERNEL_RUN_TYPE") != None and os.getenv("KAGGLE_USER_SECRETS_TOKEN") != None
 
 def _is_kaggle_handle(handle):
-    return url_pattern.match(handle) != None
+    return long_url_pattern.match(handle) != None or short_url_pattern.match(handle) != None
 
 class KaggleFileResolver(resolver.HttpResolverBase):
     def is_supported(self, handle):
         return _is_on_kaggle_notebook() and _is_kaggle_handle(handle)    
 
     def __call__(self, handle):
         m = long_url_pattern.match(handle) or short_url_pattern.match(handle)
-        return kagglehub.model_download(f"{m.group('owner')}/{m.group('model')}/{m.group('framework').lower()}/{m.group('variation')}/{m.group('version')}")
+        return kagglehub.model_download(f"{m.group('owner')}/{m.group('model')}/{m.group('framework').lower()}/{m.group('variation')}/{m.group('version')}")