whitespace

This commit is contained in:
kijai 2025-11-29 01:26:31 +02:00
parent 9a48acb211
commit 446b086ef3

View File

@ -76,7 +76,7 @@ class IndexListContextWindow(ContextWindowABC):
idx = tuple([slice(None)] * dim + [self.index_list]) idx = tuple([slice(None)] * dim + [self.index_list])
full[idx] += to_add full[idx] += to_add
return full return full
def get_region_index(self, num_regions: int) -> int: def get_region_index(self, num_regions: int) -> int:
region_idx = int(self.center_ratio * num_regions) region_idx = int(self.center_ratio * num_regions)
return min(max(region_idx, 0), num_regions - 1) return min(max(region_idx, 0), num_regions - 1)