return appropriate files

neuralmagic · Sep 6, 2023 · 00fec74 · 00fec74
1 parent 8035607
commit 00fec74
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 in file.name:
                 return file
             if isinstance(file, Directory):
                 file = file.get_file(file_name=file_name)