Remove automatic import of OpenMM scripts

openfold/np/__init__.py

@@ -11,9 +11,4 @@ for _m in _modules:
 # Avoid needlessly cluttering the global namespace
 del _files, _m, _modules
 
-from . import relax
-
-__all__.extend(["relax"])
-
-
 

setup.py

@@ -33,6 +33,7 @@ setup(
         'numpy',
         'scipy',
         'pytorch_lightning',
+        'dm-tree',
     ],
     classifiers=[
         'License :: OSI Approved :: Apache Software License',