facat commited on
Commit
c250b54
1 Parent(s): a6d7b1c
Files changed (1) hide show
  1. tlem.py +9 -1
tlem.py CHANGED
@@ -49,6 +49,14 @@ class Task:
49
  label_column: str = "answer"
50
  prompt: Optional[Callable | str] = None
51
 
 
 
 
 
 
 
 
 
52
  @cached_property
53
  def samples(self):
54
  return self.dataset[self.input_column]
@@ -59,7 +67,7 @@ class Task:
59
  *self.dataset_name
60
  if isinstance(self.dataset_name, tuple)
61
  else self.dataset_name,
62
- split=self.split
63
  )
64
  if self.prompt is not None:
65
  ds = ds.map(
 
49
  label_column: str = "answer"
50
  prompt: Optional[Callable | str] = None
51
 
52
+ @cached_property
53
+ def name(self):
54
+ return (
55
+ self.dataset_name
56
+ if isinstance(self.dataset_name, str)
57
+ else self.dataset_name[0]
58
+ ) + f"-{self.split}"
59
+
60
  @cached_property
61
  def samples(self):
62
  return self.dataset[self.input_column]
 
67
  *self.dataset_name
68
  if isinstance(self.dataset_name, tuple)
69
  else self.dataset_name,
70
+ split=self.split,
71
  )
72
  if self.prompt is not None:
73
  ds = ds.map(