init

browsergym/subtaskbench/pyproject.toml

@@ -0,0 +1,35 @@
+[build-system]
+requires = ["hatchling", "hatch-requirements-txt"]
+build-backend = "hatchling.build"
+
+[project]
+name = "browsergym-subtaskbench"
+description = "SubTaskBench benchmark for BrowserGym"
+authors = [
+    {name = "Cheng Chang"},
+    {name = "Sanjari Srivastava"},
+]
+readme = "README.md"
+requires-python = ">3.7"
+license = {text = "Apache-2.0"}
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Programming Language :: Python :: 3",
+    "Operating System :: OS Independent",
+    "Intended Audience :: Science/Research",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "License :: OSI Approved :: Apache Software License",
+]
+dynamic = ["dependencies", "version"]
+
+[project.urls]
+homepage = "https://github.com/ServiceNow/BrowserGym"
+
+[tool.hatch.version]
+path = "../core/src/browsergym/core/__init__.py"
+
+[tool.hatch.metadata.hooks.requirements_txt]
+files = ["requirements.txt"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/browsergym"]

browsergym/subtaskbench/src/browsergym/subtaskbench/__init__.py

@@ -0,0 +1,14 @@
+from browsergym.core.registration import register_task
+from . import config, task
+
+ALL_SUBTASKBENCH_TASK_IDS = []
+
+# register all SubTaskBench tasks
+for task_id in config.TASK_IDS:
+    gym_id = f"subtaskbench.{task_id}"
+    register_task(
+        gym_id,
+        task.GenericSubTaskBenchTask,
+        task_kwargs={"task_id": task_id},
+    )
+    ALL_SUBTASKBENCH_TASK_IDS.append(gym_id)