Skip to content

Commit

Permalink
flake8 (again)
Browse files Browse the repository at this point in the history
  • Loading branch information
belsten committed Nov 22, 2024
1 parent eb7727e commit 694d62c
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion sparsecoding/dictionaries.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,12 @@ def load_dictionary_from_pickle(path):
dictionary = torch.tensor(numpy_dictionary.astype(np.float32))
return dictionary


def load_bars_dictionary():
path = os.path.join(DICTIONARY_PATH, "bars", "bars-16_by_16.p")
return load_dictionary_from_pickle(path)


def load_olshausen_dictionary():
path = os.path.join(DICTIONARY_PATH, "olshausen", "olshausen-1.5x_overcomplete.p")
return load_dictionary_from_pickle(path)
return load_dictionary_from_pickle(path)

0 comments on commit 694d62c

Please sign in to comment.