[diffusion] CI: fix output-rank test fixture (#33878)
This commit is contained in:
@@ -60,7 +60,7 @@ def test_save_outputs_can_materialize_without_saving(tmp_path):
|
|||||||
|
|
||||||
def test_file_path_transport_clears_in_memory_outputs():
|
def test_file_path_transport_clears_in_memory_outputs():
|
||||||
worker = GPUWorker.__new__(GPUWorker)
|
worker = GPUWorker.__new__(GPUWorker)
|
||||||
worker.rank = 0
|
worker.is_output_rank = True
|
||||||
output_batch = OutputBatch(
|
output_batch = OutputBatch(
|
||||||
output=[object()],
|
output=[object()],
|
||||||
audio=torch.zeros(1),
|
audio=torch.zeros(1),
|
||||||
|
|||||||
Reference in New Issue
Block a user