bugfix: iter_modules needs to yield paths to container.yaml

singularityhub · Nov 7, 2022 · f069f48 · f069f48
1 parent b71fb93
commit f069f48
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/shpc/main/registry/__init__.py b/shpc/main/registry/__init__.py
@@ -189,6 +189,7 @@ def sync_from_remote(
 
         # These are modules to update
         for module in remote.iter_modules():
+            module = os.path.dirname(module)
             if name and module != name:
                 continue
 

diff --git a/shpc/main/registry/filesystem.py b/shpc/main/registry/filesystem.py
@@ -95,11 +95,7 @@ def iter_modules(self):
         yield module names
         """
         # Find modules based on container.yaml
-        for filename in shpc.utils.recursive_find(self.source, "container.yaml"):
-            module = os.path.dirname(filename)
-            if not module:
-                continue
-            yield module
+        yield from shpc.utils.recursive_find(self.source, "container.yaml")
 
     def find(self, name):
         """