Refactor DesignIP class to handle different types of designs

alsmith151 · Apr 16, 2024 · 795d87c · 795d87c
1 parent a87e523
commit 795d87c
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/seqnado/design.py b/seqnado/design.py
@@ -717,11 +717,19 @@ def from_design(
         subset_column: Optional[str] = "scale_group",
         subset_value: Optional[List[str]] = None,
     ):
-        df = (
-            design.to_dataframe()
-            .assign(sample_fullname=lambda df: df.sample_name + "_" + df.ip)
-            .set_index("sample_fullname")
-        )
+
+        if isinstance(design, Design):
+            df = (
+                design.to_dataframe()
+                .assign(sample_fullname=lambda df: df.sample_name)
+                .set_index("sample_fullname")
+            )
+        else:
+            df = (
+                design.to_dataframe()
+                .assign(sample_fullname=lambda df: df.sample_name + "_" + df.ip)
+                .set_index("sample_fullname")
+            )
 
         if subset_value:
             df = df.query(f"{subset_column} in {subset_value}")