removing unecessary code from __init__ files

openfold/model/__init__.py

-# import os
-# import glob
-# import importlib as importlib
-#
-# _files = glob.glob(os.path.join(os.path.dirname(__file__), "*.py"))
-# __all__ = [
-#     os.path.basename(f)[:-3]
-#     for f in _files
-#     if os.path.isfile(f) and not f.endswith("__init__.py")
-# ]
-# _modules = [(m, importlib.import_module("." + m, __name__)) for m in __all__]
-# for _m in _modules:
-#     globals()[_m[0]] = _m[1]
-#
-# # Avoid needlessly cluttering the global namespace
-# del _files, _m, _modules

openfold/np/__init__.py

-import os
-import glob
-import importlib as importlib
-
-_files = glob.glob(os.path.join(os.path.dirname(__file__), "*.py"))
-__all__ = [
-    os.path.basename(f)[:-3]
-    for f in _files
-    if os.path.isfile(f) and not f.endswith("__init__.py")
-]
-_modules = [(m, importlib.import_module("." + m, __name__)) for m in __all__]
-for _m in _modules:
-    globals()[_m[0]] = _m[1]
-
-# Avoid needlessly cluttering the global namespace
-del _files, _m, _modules

openfold/np/relax/__init__.py

-import os
-import glob
-import importlib as importlib
-
-_files = glob.glob(os.path.join(os.path.dirname(__file__), "*.py"))
-__all__ = [
-    os.path.basename(f)[:-3]
-    for f in _files
-    if os.path.isfile(f) and not f.endswith("__init__.py")
-]
-_modules = [(m, importlib.import_module("." + m, __name__)) for m in __all__]
-for _m in _modules:
-    globals()[_m[0]] = _m[1]
-
-# Avoid needlessly cluttering the global namespace
-del _files, _m, _modules

openfold/utils/__init__.py

-import os
-import glob
-import importlib as importlib
-
-from . import kernel
-
-_files = glob.glob(os.path.join(os.path.dirname(__file__), "*.py"))
-__all__ = [
-    os.path.basename(f)[:-3]
-    for f in _files
-    if os.path.isfile(f) and not f.endswith("__init__.py")
-] + ["kernel"]
-_modules = [(m, importlib.import_module("." + m, __name__)) for m in __all__]
-for _m in _modules:
-    globals()[_m[0]] = _m[1]
-
-# Avoid needlessly cluttering the global namespace
-del _files, _m, _modules

thread_sequence.py

@@ -38,9 +38,9 @@ def main(args):
 
     config = model_config(args.config_preset)
 
-
-    #TODO make configurable
-    random_seed = random.randrange(2 ** 32)
+    random_seed = args.data_random_seed
+    if random_seed is None:
+        random_seed = random.randrange(2**32)
 
     numpy.random.seed(random_seed)
     torch.manual_seed(random_seed + 1)
@@ -150,6 +150,10 @@ if __name__ == "__main__":
                  of the pLDDT itself"""
     )
 
+    parser.add_argument(
+        "--data_random_seed", type=str, default=None
+    )
+
     add_data_args(parser)
 
     args = parser.parse_args()