Update .gihub/imgs ProteinGym

niklases · Aug 5, 2024 · 90e113b · 90e113b
1 parent b6863b2
commit 90e113b
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/.github/imgs/multi_point_mut_performance_PGym_DCA.png b/.github/imgs/multi_point_mut_performance_PGym_DCA.png
diff --git a/.github/imgs/single_point_mut_performance_PGyM_DCA.png b/.github/imgs/single_point_mut_performance_PGyM_DCA.png
diff --git a/scripts/ProteinGym_runs/run_performance_tests_proteingym_data.py b/scripts/ProteinGym_runs/run_performance_tests_proteingym_data.py
@@ -111,7 +111,7 @@ def plot_performance(mut_data, plot_name, mut_sep=':'):
             f'0' + r'$\rightarrow$' + f'{len(fitnesses)}', 
             color='tab:blue', size=4, ha='right'
         ))
-        assert len(x_dca) == len(fitnesses) == len(variants) == len(sequences)
+        assert len(x_dca) == len(fitnesses) == len(variants) == len(sequences) == len(y_pred)
         hybrid_perfs = []
         for i_t, train_size in enumerate([25, 50, 75, 100, 200]):
             try: