Skip to content

Commit

Permalink
bug fix in sequence optimziation
Browse files Browse the repository at this point in the history
  • Loading branch information
ryanemenecker committed Nov 22, 2024
1 parent 4403319 commit 125c030
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion goose/sequence_optimization.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,7 @@ def set_fixed_ranges(self, ranges: List[Tuple[int, int]]):

def set_optimization_params(self, max_iterations: int = None, tolerance: float = None,
window_size: int = None, num_shuffles: int = None,
shuffle_interval: int = None):
shuffle_interval: int = None, just_shuffle: bool = None):
if max_iterations is not None:
self.max_iterations = max_iterations
if tolerance is not None:
Expand All @@ -189,6 +189,8 @@ def set_optimization_params(self, max_iterations: int = None, tolerance: float =
self.num_shuffles = num_shuffles
if shuffle_interval is not None:
self.shuffle_interval = shuffle_interval
if just_shuffle is not None:
self.just_shuffle = just_shuffle

def set_initial_sequence(self, sequence: str):
assert len(sequence) == self.target_length, f"Initial sequence length must match target length ({self.target_length})"
Expand Down

0 comments on commit 125c030

Please sign in to comment.