revert repr changes

relbench/base/dataset.py

@@ -41,13 +41,7 @@ def __init__(
  self.cache_dir = cache_dir
 
  def __repr__(self) -> str:
- return (
- f"{self.__class__.__name__}(\n"
- f"val_timestamp={self.val_timestamp},\n"
- f"test_timestamp={self.test_timestamp},\n"
- # f"cache_dir={self.cache_dir},\n"
- f")"
- )
+ return f"{self.__class__.__name__}()"
 
  def validate_and_correct_db(self, db):
  r"""Validate and correct input db in-place.
@@ -100,7 +94,7 @@ def get_db(self, upto_test_timestamp=True) -> Database:
  else:
  print("Making Database object from scratch...")
  print(
- "(You can also use `get_dataset(..., download=True)`"
+ "(You can also use `get_dataset(..., download=True)` "
  "for datasets prepared by the RelBench team.)"
  )
  tic = time.time()

relbench/base/task_base.py

@@ -78,15 +78,7 @@ def __init__(
  )
 
  def __repr__(self) -> str:
- return (
- f"{self.__class__.__name__}(\n"
- f"dataset={self.dataset},\n"
- # f"cache_dir={self.cache_dir},\n"
- f"task_type={self.task_type},\n"
- f"timedelta={self.timedelta},\n"
- # f"num_eval_timestamps={self.num_eval_timestamps},\n"
- # f"metrics={self.metrics},\n"
- )
+ return f"{self.__class__.__name__}(dataset={repr(self.dataset)})"
 
  def make_table(
  self,