File size: 587 Bytes
36ba3ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import pkg_resources
import os
import voyager.utils as U
def load_control_primitives_context(primitive_names=None):
package_path = pkg_resources.resource_filename("voyager", "")
if primitive_names is None:
primitive_names = [
primitive[:-3]
for primitive in os.listdir(f"{package_path}/control_primitives_context")
if primitive.endswith(".js")
]
primitives = [
U.load_text(f"{package_path}/control_primitives_context/{primitive_name}.js")
for primitive_name in primitive_names
]
return primitives
|