[RLlib] Broken restore from remote - Add missing FileSystem argument (#58324)

Daraan · web-flow · commit 7271b0cc5b45 · 2025-11-07T20:10:26.000+01:00
diff --git a/rllib/algorithms/algorithm.py b/rllib/algorithms/algorithm.py
@@ -343,7 +343,7 @@ def from_checkpoint(
                 new="Algorithm.from_checkpoint(path=...)",
                 error=True,
             )
-        checkpoint_info = get_checkpoint_info(path)
+        checkpoint_info = get_checkpoint_info(path, filesystem)
 
         # New API stack -> Use Checkpointable's default implementation.
         if checkpoint_info["checkpoint_version"] >= version.Version("2.0"):

Original file line number	Diff line number	Diff line change
`@@ -343,7 +343,7 @@ def from_checkpoint(`
`343`	`343`	`new="Algorithm.from_checkpoint(path=...)",`
`344`	`344`	`error=True,`
`345`	`345`	`)`
`346`		`- checkpoint_info = get_checkpoint_info(path)`
	`346`	`+ checkpoint_info = get_checkpoint_info(path, filesystem)`
`347`	`347`
`348`	`348`	`# New API stack -> Use Checkpointable's default implementation.`
`349`	`349`	`if checkpoint_info["checkpoint_version"] >= version.Version("2.0"):`