Added __all__ from direct_collocation and utils to package namespace.

csu-hmc · Jul 23, 2024 · 0248f29 · 0248f29
1 parent bd0ef0e
commit 0248f29
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/opty/__init__.py b/opty/__init__.py
@@ -1,3 +1,5 @@
 #!/usr/bin/env python
 
 from .version import __version__
+from .direct_collocation import *
+from .utils import *
diff --git a/opty/utils.py b/opty/utils.py
@@ -29,6 +29,11 @@
                                     import_kwargs={'fromlist': ['']},
                                     catch=(RuntimeError,))
 
+__all__ = [
+    'parse_free',
+    'create_objective_function',
+]
+
 
 def _forward_jacobian(expr, wrt):