feat: adding filename to Save/Load models

VirtualBrainLab · Apr 4, 2024 · e79b3ef · e79b3ef
1 parent 9faec06
commit e79b3ef
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/vbl_aquarium/models/dock.py b/src/vbl_aquarium/models/dock.py
@@ -18,10 +18,12 @@ class DownloadModel(VBLBaseModel):
 
 # Models for sending save/load messages
 class SaveModel(VBLBaseModel):
-    bucket: str
-    password: str
+    filename: str
+    bucket: str = ''
+    password: str = ''
 
 
 class LoadModel(VBLBaseModel):
-    bucket: str
-    password: str
+    filename: str
+    bucket: str = ''
+    password: str = ''