@@ -31,8 +31,11 @@ async def read_file(self, filename: str) -> bytes:
3131 """
3232 Read a binary file from the filesystem.
3333
34- :param filename: Relative path to the file.
35- :return: Binary content of the file.
34+ Args:
35+ filename: Relative path to the file.
36+
37+ Returns:
38+ Binary content of the file.
3639 """
3740
3841 def stdlib_read ():
@@ -56,17 +59,23 @@ def get_full_path(self, filename: str) -> str:
5659 """
5760 Get the full path for a given filename within the project folder.
5861
59- :param filename: Name of the file.
60- :return: Full path to the file.
62+ Args:
63+ filename: Name of the file.
64+
65+ Returns:
66+ Full path to the file.
6167 """
6268
6369 async def save_file (self , filename : str , content : bytes ) -> str :
6470 """
6571 Save a binary file to the filesystem under the project directory.
6672
67- :param filename: Name of the file to save.
68- :param content: Binary content of the file.
69- :return: The path where the file was saved.
73+ Args:
74+ filename: Name of the file to save.
75+ content: Binary content of the file.
76+
77+ Returns:
78+ The path where the file was saved.
7079 """
7180
7281 def stdlib_write ():
@@ -87,7 +96,8 @@ async def delete_file(self, filename: str) -> None:
8796 """
8897 Delete a binary file from the filesystem.
8998
90- :param filename: Name of the file to delete.
99+ Args:
100+ filename: Name of the file to delete.
91101 """
92102
93103 def stdlib_delete ():
@@ -122,8 +132,11 @@ def get_snapshot_path(self, snapshot_id: str | UUID) -> str:
122132 """
123133 Get the full path for a dataset snapshot.
124134
125- :param snapshot_id: ID of the snapshot.
126- :return: Full path to the snapshot file.
135+ Args:
136+ snapshot_id: ID of the snapshot.
137+
138+ Returns:
139+ Full path to the snapshot file.
127140 """
128141 return self .get_full_path (f"{ snapshot_id } .parquet" )
129142
@@ -149,7 +162,8 @@ def model_folder_path(self) -> str:
149162 """
150163 Get the folder path for models.
151164
152- :return: Folder path for models.
165+ Returns:
166+ Folder path for models.
153167 """
154168 return os .path .join (self .project_folder_path , self ._model_id )
155169
@@ -171,9 +185,12 @@ def get_weights_file_path(self, format: ExportType, name: str) -> str:
171185 """
172186 Read a weights file from the model folder.
173187
174- :param format: Format of the model (e.g., ExportType.OPENVINO).
175- :param name: Name of the weights to read.
176- :return: path of the weights file.
188+ Args:
189+ format: Format of the model (e.g., ExportType.OPENVINO).
190+ name: Name of the weights to read.
191+
192+ Returns:
193+ path of the weights file.
177194 """
178195 return os .path .join (self .model_folder_path , "weights" , format , name )
179196
@@ -191,17 +208,21 @@ def model_export_folder_path(self) -> str:
191208 """
192209 Get the folder path for model exports.
193210
194- :return: Folder path for model exports.
211+ Returns:
212+ Folder path for model exports.
195213 """
196214 return os .path .join (self .project_folder_path , self ._model_id )
197215
198216 def get_model_export_path (self , model_name : str , export_params : ExportParameters ) -> str :
199217 """
200218 Get the full path for a dataset snapshot.
201219
202- :param model_name: name of the model
203- :param export_params: model export parameters
204- :return: Full path to the model export zip file.
220+ Args:
221+ model_name: name of the model
222+ export_params: model export parameters
223+
224+ Returns:
225+ Full path to the model export zip file.
205226 """
206227 compression_suffix = f"_{ export_params .compression .value } " if export_params .compression else ""
207228 filename = f"{ model_name } _{ export_params .format .value } { compression_suffix } .zip"
0 commit comments