From afceaf1dfe0d16733d2237e1e10d30e0b64b9da3 Mon Sep 17 00:00:00 2001
From: Philipp Holl <philipp.holl@tum.de>
Date: Fri, 12 Jan 2024 15:14:53 +0100
Subject: [PATCH] [io] Fix Scene.at()

---
 phi/field/_scene.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/phi/field/_scene.py b/phi/field/_scene.py
index 006f651e4..a4f5ebfae 100644
--- a/phi/field/_scene.py
+++ b/phi/field/_scene.py
@@ -202,7 +202,7 @@ def at(directory: Union[str, tuple, typing_list, math.Tensor, 'Scene'], id: Unio
             return directory
         if isinstance(directory, (tuple, list)):
             directory = math.wrap(directory, batch('scenes'))
-        directory = math.map(lambda d: expanduser(d), math.wrap(directory))
+        directory = math.wrap(math.map(lambda d: expanduser(d), directory))
         if isinstance(id, int) and id < 0:
             assert directory.shape.volume == 1
             scenes = Scene.list(directory.native())