Skip to content

Commit 5581d05

Browse files
committed
[Notebook] Fix bug
1 parent 5faccc4 commit 5581d05

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

notebooks/Simulation_and_training_in_VMAS_and_BenchMARL.ipynb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2160,8 +2160,10 @@
21602160
" device: DEVICE_TYPING,\n",
21612161
") -> Callable[[], EnvBase]:\n",
21622162
" config = copy.deepcopy(self.config)\n",
2163-
" if self is VmasTask.NAVIGATION: # This is the only modification we make ....\n",
2164-
" scenario = MyScenario() # .... ends here\n",
2163+
" if (hasattr(self, \"name\") and self.name is \"NAVIGATION\") or (\n",
2164+
" self is VmasTask.NAVIGATION\n",
2165+
" ): # This is the only modification we make ....\n",
2166+
" scenario = MyScenario() # .... ends here\n",
21652167
" else:\n",
21662168
" scenario = self.name.lower()\n",
21672169
" return lambda: VmasEnv(\n",

0 commit comments

Comments
 (0)