Update __init__.py

This commit is contained in:
向劲宇 2023-09-27 02:15:35 +08:00 committed by GitHub
parent b090a628a0
commit 8f324ba91b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -4,15 +4,15 @@ import voyager.utils as U
def load_control_primitives_context(primitive_names=None):
package_path = pkg_resources.resource_filename("voyager", "")
package_path = pkg_resources.resource_filename("metagpt", "")
if primitive_names is None:
primitive_names = [
primitive[:-3]
for primitive in os.listdir(f"{package_path}/control_primitives_context")
for primitive in os.listdir(f"{package_path}/actions/minecraft/control_primitives_context")
if primitive.endswith(".js")
]
primitives = [
U.load_text(f"{package_path}/control_primitives_context/{primitive_name}.js")
U.load_text(f"{package_path}/actions/minecraft/control_primitives_context/{primitive_name}.js")
for primitive_name in primitive_names
]
return primitives