Replace push_sample by create_sample

This commit is contained in:
2021-07-27 09:25:40 -05:00
parent 4224586d10
commit 962707e8b7
3 changed files with 7 additions and 6 deletions

View File

@@ -137,9 +137,9 @@ class PickleGzInstance(Instance):
return self.instance.get_samples()
@overrides
def push_sample(self, sample: Sample) -> None:
def create_sample(self) -> Sample:
assert self.instance is not None
self.instance.push_sample(sample)
return self.instance.create_sample()
def write_pickle_gz(obj: Any, filename: str) -> None: