docs: update docstring

adaptyvbio · Dec 28, 2023 · 46822ef · 46822ef
1 parent 84676be
commit 46822ef
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/proteinflow/data/torch.py b/proteinflow/data/torch.py
@@ -908,8 +908,9 @@ def set_cdr(self, cdr):
 
         Parameters
         ----------
-        cdr : {"H1", "H2", "H3", "L1", "L2", "L3"}
-            The CDR to be iterated over. Set to `None` to go back to iterating over all chains.
+        cdr : list | str | None
+            The CDR to be iterated over (choose from H1, H2, H3, L1, L2, L3).
+            Set to `None` to go back to iterating over all chains.
 
         """
         if not self.sabdab: