benchmark group bug fix

README.md

@@ -23,7 +23,7 @@ TDC is an open-source initiative. To get involved, join the [Slack Workspace](ht
 
 
 ## Updates
-- `0.1.7`: Streamlined the leaderboard programming frameworks! Checkout [here](https://tdcommons.ai/benchmark/overview/)!
+- `0.1.8`: Streamlined and simplified the leaderboard programming frameworks! Now, you can submit a result for a single dataset! Checkout [here](https://tdcommons.ai/benchmark/overview/)!
 - TDC white paper is alive on [arXiv](https://arxiv.org/abs/2102.09548)!
 - `0.1.6`: Released the second leaderboard on drug combination screening prediction! Checkout [here](https://tdcommons.ai/benchmark/drugcombo_group/)!
 - `0.1.5`: Added four realistic oracles from docking scores and synthetic accessibility! Checkout [here](https://tdcommons.ai/functions/oracles/)!

tdc/benchmark.py

@@ -138,7 +138,7 @@ def get(self, benchmark):
 		elif self.file_format == 'pkl':
 			train = pd.read_pickle(os.path.join(data_path, 'train_val.pkl'))
 			test = pd.read_pickle(os.path.join(data_path, 'test.pkl'))
-		return {'train': train, 'test': test, 'name': dataset}
+		return {'train_val': train, 'test': test, 'name': dataset}
 
 	def evaluate(self, pred, true = None, benchmark = None):
 		if true is None:

tdc/version.py

@@ -19,4 +19,4 @@
 # Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
 # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
 #
-__version__ = '0.1.7'  # pragma: no cover
+__version__ = '0.1.8'  # pragma: no cover