fix: debug mode - fix get the number of samples in testtuple
@@ -274,7 +274,7 @@ class Worker:
if not self._db.is_local(dataset.key):
command += " --fake-data"
- command += f" --n-fake-samples {len(objective.test_dataset.data_sample_keys)}"
+ command += f" --n-fake-samples {len(tuple_.dataset.keys)}"
if tuple_.traintuple_type == schemas.Type.Traintuple \
or tuple_.traintuple_type == schemas.Type.Aggregatetuple:
@@ -321,7 +321,7 @@ class Worker:
command = f"--fake-data-mode {METRICS_NO_FAKE_Y}"
else:
command = f"--fake-data-mode {METRICS_FAKE_Y}"
- command += f" --n-fake-samples {len(objective.test_dataset.data_sample_keys)}"
+ command += f" --n-fake-samples {len(tuple_.dataset.keys)}"
container_name = DOCKER_METRICS_TAG
logs_predict = self._spawner.spawn(
