Test name fix

svalkiers · Aug 3, 2022 · e6ce0f8 · e6ce0f8
1 parent 266980a
commit e6ce0f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/clustcr/clustering/clustering.py b/clustcr/clustering/clustering.py
@@ -332,7 +332,7 @@ def batch_precluster(self, cdr3: pd.Series, name=''):
         for index, row in clustered.clusters_df.iterrows():
             filename = join(Clustering.BATCH_TMP_DIRECTORY, str(row['cluster']))
             with open(filename, 'a') as f:
-                f.write(f'{row["CDR3"]},{name}\n')
+                f.write(f'{row["junction_aa"]},{name}\n')
 
     def batch_cluster(self, calc_cluster_matrix=False):
         """

diff --git a/test/test_clustering.py b/test/test_clustering.py
@@ -74,7 +74,7 @@ def test_batch_clustering_multiprocessing(self):
 
     def test_matrix(self):
         vdj = datasets.vdjdb_beta()
-        max_sequence_size = vdj.str.len().max()
+        max_sequence_size = vdj.junction_aa.str.len().max()
         train = vdj.sample(2000)
         times = 3
         size_per_time = 3000