Merge pull request #631 from Isaac-Flath/NestedAttr

fastai · Oct 2, 2024 · cc75f59 · cc75f59
2 parents 2ff4004 + 7044728
commit cc75f59
Show file tree

Hide file tree

Showing 2 changed files with 109 additions and 54 deletions.
diff --git a/fastcore/basics.py b/fastcore/basics.py
@@ -714,8 +714,11 @@ def only(o):
 def nested_attr(o, attr, default=None):
     "Same as `getattr`, but if `attr` includes a `.`, then looks inside nested objects"
     try:
-        for a in attr.split("."): o = getattr(o, a)
-    except AttributeError: return default
+        for a in attr.split("."): 
+            if hasattr(o,a): o = getattr(o, a)
+            else: o = o[a]
+    except (AttributeError, KeyError,IndexError, TypeError, ValueError):
+        return default
     return o
 
 # %% ../nbs/01_basics.ipynb