Forgot to sort first_run_files too

FNNDSC · Mar 1, 2024 · 38c0937 · 38c0937
1 parent 4356742
commit 38c0937
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/visualdataset/__init__.py b/visualdataset/__init__.py
@@ -9,4 +9,4 @@
 |_|
 """
 
-__version__ = '0.2.2'
+__version__ = '0.2.3'
diff --git a/visualdataset/manifest.py b/visualdataset/manifest.py
@@ -58,12 +58,20 @@ class VisualDatasetManifest(BaseModel):
     __pydantic_config__ = ConfigDict(extra='forbid')
 
     def sort(self) -> Self:
+        sorted_tags = {
+            k: sorted(v)
+            for k, v in sorted(self.tags.items())
+        }
+        sorted_files = sorted(self.files, key=_get_path)
+        sorted_file_paths = [f.path for f in sorted_files]
+        sorted_first_run_files = [
+            sorted_file_paths.index(self.files[i].path)
+            for i in self.first_run_files
+        ]
         return self.model_copy(update={
-            'tags': {
-                k: sorted(v)
-                for k, v in sorted(self.tags.items())
-            },
-            'files': sorted(self.files, key=_get_path)
+            'tags': sorted_tags,
+            'files': sorted_files,
+            'first_run_files': sorted_first_run_files
         })