Return appropriate files (#363)

neuralmagic · Sep 11, 2023 · 9d9b2ba · 9d9b2ba
1 parent 05e55e5
commit 9d9b2ba
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 == file_name:
+            if file.name == os.path.basename(file_name):
                 return file
             if isinstance(file, Directory):
                 file = file.get_file(file_name=file_name)