File size: 566 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(primitive_names=None):
package_path = pkg_resources.resource_filename("voyager", "")
if primitive_names is None:
primitive_names = [
primitives[:-3]
for primitives in os.listdir(f"{package_path}/control_primitives")
if primitives.endswith(".js")
]
primitives = [
U.load_text(f"{package_path}/control_primitives/{primitive_name}.js")
for primitive_name in primitive_names
]
return primitives
|