linting spec2vec

iomega · Aug 14, 2024 · 10e34ab · 10e34ab
1 parent 0194a81
commit 10e34ab
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/spec2vec/Spec2Vec.py b/spec2vec/Spec2Vec.py
@@ -177,12 +177,11 @@ def matrix(self, references: Union[List[SpectrumDocument], List[Spectrum]],
 
         if array_type == "numpy":
             return spec2vec_similarity
-        elif array_type == "sparse":
+        if array_type == "sparse":
             sparse = StackedSparseArray(n_rows, n_cols)
             sparse.add_dense_matrix(spec2vec_similarity, "")
             return sparse
-        else:
-            raise NotImplementedError("Only 'numpy' and 'sparse' array types are supported.")
+        raise NotImplementedError("Only 'numpy' and 'sparse' array types are supported.")
 
     @staticmethod
     def _get_word_decimals(model):