fix

neuralmagic · Sep 8, 2023 · e35a4e2 · e35a4e2
1 parent 77ec763
commit e35a4e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparsezoo/objects/directory.py b/src/sparsezoo/objects/directory.py
@@ -212,7 +212,7 @@ def get_file(self, file_name: str) -> Optional[File]:
         for file in self.files:
             if file is None:
                 continue
-            if file_name == os.path.basename(file_name):
+            if file.name == os.path.basename(file_name):
                 return file
             if isinstance(file, Directory):
                 file = file.get_file(file_name=file_name)