Skip to content
Prev Previous commit
Next Next commit
Update loreal_poc/dataloaders/base.py
  • Loading branch information
rabah-khalek authored Jan 4, 2024
commit 566d1fb616ac03c3342c8a061e3e84e49f16e5ce
2 changes: 1 addition & 1 deletion loreal_poc/dataloaders/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def __init__(self, name: str, batch_size: int = 1):
raise ValueError(f"Batch size must be a strictly positive integer: {self.batch_size}")

@property
def name(self) -> np.ndarray:
def name(self) -> str:
return self._name

def __iter__(self):
Expand Down