Skip to content

Commit 3d9a1d8

Browse files
committed
Fix bug
1 parent 21f0a60 commit 3d9a1d8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ms2query/ms2query_development/ReferenceLibrary.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def load_from_directory(cls, library_file_directory) -> "ReferenceLibrary":
6868
reference_embeddings_file = library_file_directory / cls.embedding_file_name
6969
top_k_tanimoto_scores_file = library_file_directory / cls.top_k_tanimoto_scores_file_name
7070
reference_metadata_file = library_file_directory / cls.reference_metadata_file_name
71-
ms2deepscore_model_file_name = library_file_directory = cls.ms2deepscore_model_file_name
71+
ms2deepscore_model_file_name = library_file_directory / cls.ms2deepscore_model_file_name
7272
return cls.load_from_files(
7373
ms2deepscore_model_file_name, reference_embeddings_file, top_k_tanimoto_scores_file, reference_metadata_file
7474
)
@@ -99,6 +99,8 @@ def create_from_spectra(
9999
"""Creates all the files needed for MS2Query and stores them"""
100100
if store_file_directory is None:
101101
store_file_directory = Path(ms2deepscore_model_file_name).parent
102+
else:
103+
store_file_directory = Path(store_file_directory)
102104
if store_files:
103105
# Check the files don't exist yet
104106
for file in (

0 commit comments

Comments
 (0)