changes

Author: mohit-singlestore

singlestoredb/fusion/handlers/models.py

@@ -10,7 +10,7 @@
 from .utils import get_file_space
 
 
-class ShowModelsHandler(ShowFilesHandler):
+class ShowModelFilesHandler(ShowFilesHandler):
     """
     SHOW MODEL FILES
         [ at_path ] [ <like> ]
@@ -73,10 +73,10 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return super().run(params)
 
 
-ShowModelsHandler.register(overwrite=True)
+ShowModelFilesHandler.register(overwrite=True)
 
 
-class UploadModelHandler(SQLHandler):
+class UploadModelFileHandler(SQLHandler):
     """
     UPLOAD MODEL FILE model_name
         FROM local_path [ overwrite ];
@@ -143,10 +143,10 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return None
 
 
-UploadModelHandler.register(overwrite=True)
+UploadModelFileHandler.register(overwrite=True)
 
 
-class DownloadModelHandler(SQLHandler):
+class DownloadModelFileHandler(SQLHandler):
     """
     DOWNLOAD MODEL FILE model_name
         [ local_path ]
@@ -211,10 +211,10 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return None
 
 
-DownloadModelHandler.register(overwrite=True)
+DownloadModelFileHandler.register(overwrite=True)
 
 
-class DropModelsHandler(SQLHandler):
+class DropModelFileHandler(SQLHandler):
     """
     DROP MODEL FILE model_name;
 
@@ -247,4 +247,4 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return None
 
 
-DropModelsHandler.register(overwrite=True)
+DropModelFileHandler.register(overwrite=True)