Skip to content

Commit

Permalink
Black formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
raphaelvallat committed Dec 30, 2022
1 parent e316cf4 commit 3320530
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions yasa/staging.py
Original file line number Diff line number Diff line change
Expand Up @@ -422,6 +422,7 @@ def predict(self, path_to_model="auto"):
returned as a :py:class:`yasa.Hypnogram` instance.
"""
from yasa.hypno import Hypnogram

if not hasattr(self, "_features"):
self.fit()
# Load and validate pre-trained classifier
Expand Down

0 comments on commit 3320530

Please sign in to comment.