Skip to content

Commit

Permalink
Fix linting.
Browse files Browse the repository at this point in the history
  • Loading branch information
shyuep committed Nov 6, 2024
1 parent e5a432d commit a37b86e
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions src/maml/base/_describer.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def __init__(self, **kwargs):
for k in kwargs:
if k not in allowed_kwargs:
raise TypeError(f"{k} not allowed as kwargs")
memory = kwargs.get("memory", None)
memory = kwargs.get("memory")
if isinstance(memory, bool):
memory = tempfile.mkdtemp()
logger.info(f"Created temporary directory {memory}")
Expand All @@ -80,7 +80,7 @@ def __init__(self, **kwargs):
n_jobs = cpu_count()
logger.info(f"Using {n_jobs} jobs for computation")
self.n_jobs = n_jobs
self.feature_batch = get_feature_batch(kwargs.get("feature_batch", None))
self.feature_batch = get_feature_batch(kwargs.get("feature_batch"))

def fit(self, x: Any, y: Any = None) -> BaseDescriber:
"""
Expand Down
2 changes: 1 addition & 1 deletion src/maml/utils/_value_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ def __init__(self, value_start: float, value_end: float = 0.0, max_steps: int =
**kwargs: captures anything else.
"""
self.value_start = value_start
rate = kwargs.get("rate", None)
rate = kwargs.get("rate")
if rate is None:
self.rate = (value_end - self.value_start) / max_steps
else:
Expand Down

0 comments on commit a37b86e

Please sign in to comment.