Skip to content

Commit 951b03b

Browse files
Denis Barakhtanov0xE0F
authored andcommitted
Fix checkpoint ftest
Signed-off-by: Denis Barakhtanov <dbarahtanov@enakta.com>
1 parent 7512235 commit 951b03b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/tests/ftest/pytorch/checkpoint.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -84,21 +84,21 @@ def test_checkpoint_nested_directories(self):
8484
:avocado: tags=PytorchCheckpointTest,test_checkpoint_nested_directories
8585
"""
8686

87-
pool = self.get_pool().identifier
88-
container = self.get_container(pool).identifier
87+
pool = self.get_pool()
88+
container = self.get_container(pool)
8989

9090
d1, d2 = str(uuid.uuid4()), str(uuid.uuid4())
9191
files = ["/file.pt", f"/{d1}/file.pt", f"/{d1}/{d2}/file.pt"]
9292

9393
# by default parent directories should be created
94-
with Checkpoint(pool, container) as pt:
94+
with Checkpoint(pool.identifier, container.identifier) as pt:
9595
for name in files:
9696
with pt.writer(name) as w:
9797
w.write(os.urandom(4096))
9898

9999
# ensure that it fails with expected exception
100100
try:
101-
with Checkpoint(pool, container) as pt:
101+
with Checkpoint(pool.identifier, container.identifier) as pt:
102102
fname = f"/{str(uuid.uuid4())}/file.pt"
103103
with pt.writer(fname, ensure_path=False) as w:
104104
w.write(os.urandom(4096))

0 commit comments

Comments
 (0)