make PotcarSummaryStats subscriptable

esoteric-ephemera · esoteric-ephemera · commit 373cd9af4606 · 2025-01-28T15:58:51.000-08:00
diff --git a/emmet-core/emmet/core/vasp/calculation.py b/emmet-core/emmet/core/vasp/calculation.py
@@ -98,6 +98,10 @@ class PotcarSummaryStats(BaseModel):
         ),
     )
 
+    def __getitem__(self, name: str) -> Any:
+        """Make attributes subscriptable."""
+        return getattr(self, name)
+
 
 class PotcarSpec(BaseModel):
     """Document defining a VASP POTCAR specification."""

Original file line number	Diff line number	Diff line change
`@@ -98,6 +98,10 @@ class PotcarSummaryStats(BaseModel):`
`98`	`98`	`),`
`99`	`99`	`)`
`100`	`100`
	`101`	`+ def __getitem__(self, name: str) -> Any:`
	`102`	`+ """Make attributes subscriptable."""`
	`103`	`+ return getattr(self, name)`
	`104`	`+`
`101`	`105`
`102`	`106`	`class PotcarSpec(BaseModel):`
`103`	`107`	`"""Document defining a VASP POTCAR specification."""`