Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Chris Broz <Chris.Broz@ucsf.edu>
  • Loading branch information
samuelbray32 and CBroz1 authored Sep 16, 2024
1 parent f685dad commit 601ed56
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion src/spyglass/utils/dj_merge_tables.py
Original file line number Diff line number Diff line change
Expand Up @@ -835,8 +835,13 @@ def super_delete(self, warn=True, *args, **kwargs):
super().delete(*args, **kwargs)

@classmethod
def extract_merge_id(cls, restriction):
def extract_merge_id(cls, restriction) -> Union[dict, list]:
"""Utility function to extract merge_id from a restriction
Removes all other restricted attributes, and defaults to a
universal set (either empty dict or True) when there is no
merge_id present in the input, relying on parent func to
restrict on secondary or part-parent key(s).
Parameters
----------
Expand Down

0 comments on commit 601ed56

Please sign in to comment.