[datasets] Add a new Dataset class.

Issue facebookresearch#45.
bwasti · Aug 3, 2021 · 17fed3e · 17fed3e
1 parent d373b67
commit 17fed3e
Show file tree

Hide file tree

Showing 8 changed files with 474 additions and 23 deletions.
diff --git a/compiler_gym/bin/BUILD b/compiler_gym/bin/BUILD
@@ -22,7 +22,7 @@ py_binary(
  srcs = ["datasets.py"],
  visibility = ["//visibility:public"],
  deps = [
- "//compiler_gym/datasets:dataset",
+ "//compiler_gym/datasets",
  "//compiler_gym/envs",
  "//compiler_gym/util",
  "//compiler_gym/util/flags:env_from_flags",

diff --git a/compiler_gym/datasets/BUILD b/compiler_gym/datasets/BUILD
@@ -18,12 +18,3 @@ py_library(
  "//compiler_gym/util",
  ],
 )
-
-py_library(
- name = "dataset",
- srcs = ["dataset.py"],
- visibility = ["//compiler_gym:__subpackages__"],
- deps = [
- "//compiler_gym/util",
- ],
-)
diff --git a/compiler_gym/datasets/__init__.py b/compiler_gym/datasets/__init__.py
@@ -9,6 +9,8 @@
  BenchmarkSource,
 )
 from compiler_gym.datasets.dataset import (
+ Dataset,
+ DatasetInitError,
  LegacyDataset,
  activate,
  deactivate,
@@ -21,6 +23,8 @@
  "Benchmark",
  "BenchmarkInitError",
  "BenchmarkSource",
+ "Dataset",
+ "DatasetInitError",
  "deactivate",
  "delete",
  "LegacyDataset",