[loop_tool] Remove unused site data path from datasets.

facebookresearch · Dec 14, 2021 · b797c33 · b797c33
1 parent d436457
commit b797c33
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/compiler_gym/envs/loop_tool/__init__.py b/compiler_gym/envs/loop_tool/__init__.py
@@ -9,7 +9,7 @@
 from compiler_gym.datasets import Benchmark, Dataset, benchmark
 from compiler_gym.spaces import Reward
 from compiler_gym.util.registration import register
-from compiler_gym.util.runfiles_path import runfiles_path, site_data_path
+from compiler_gym.util.runfiles_path import runfiles_path
 
 LOOP_TOOL_SERVICE_BINARY: Path = runfiles_path(
     "compiler_gym/envs/loop_tool/service/compiler_gym-loop_tool-service"
@@ -56,7 +56,6 @@ def __init__(self, *args, **kwargs):
             name="benchmark://loop_tool-cuda-v0",
             license="MIT",
             description="loop_tool dataset",
-            site_data_base=site_data_path("loop_tool_dataset"),
         )
 
     def benchmark_uris(self) -> Iterable[str]:
@@ -72,7 +71,6 @@ def __init__(self, *args, **kwargs):
             name="benchmark://loop_tool-cpu-v0",
             license="MIT",
             description="loop_tool dataset",
-            site_data_base=site_data_path("loop_tool_dataset"),
         )
 
     def benchmark_uris(self) -> Iterable[str]: