Skip to content

Commit

Permalink
Use set for member_number split
Browse files Browse the repository at this point in the history
  • Loading branch information
HCookie committed Dec 11, 2024
1 parent 82580b1 commit de51c45
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/ai_models_gencast/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def __init__(self, **kwargs):
]

if isinstance(self.member_number, str):
self.member_number = list(map(int, self.member_number.split(",")))
self.member_number = list(set(map(int, self.member_number.split(","))))
elif isinstance(self.member_number, int):
self.member_number = [int(self.member_number)]
elif self.member_number is None:
Expand Down

0 comments on commit de51c45

Please sign in to comment.