Merge pull request #351 from scott-huberty/derivatives_repr

ENH: better repr for Derivatives class

Author: mgxd

nibabies/utils/bids.py

@@ -67,7 +67,9 @@ def __init__(self, bids_root: Path | str, spec: dict | Path | str | None = None,
             setattr(self, name, None)
 
     def __repr__(self):
-        return '\n'.join([name for name in self.names if getattr(self, name)])
+        output = ["Derivatives |"]
+        output.extend([f"  {attr}: {getattr(self, attr)}" for attr in self.names])
+        return '\n'.join(output)
 
     def __contains__(self, val: str):
         return val in self.names

nibabies/utils/tests/test_bids.py

@@ -99,6 +99,7 @@ def test_derivatives(
     assert derivatives.aseg is None
     assert derivatives.t1w_aseg is None
     assert derivatives.t2w_aseg is None
+    assert "t1w_mask" in repr(derivatives)
 
     derivatives.populate(deriv_dir, subject_id='01')
     if mask:
@@ -111,3 +112,5 @@ def test_derivatives(
         assert derivatives.references[aseg]
     else:
         assert derivatives.aseg == None
+    if t1w_mask:
+        assert "sub-01_space-T1w_desc-brain_mask.nii.gz" in repr(derivatives)