reason_text,
)
return compose_sample_weights(
- base_weights, None, logger=logger, context=context
- )
+ base_weights, None, logger=logger, context=context
+ )
case _:
assert_never(policy)
train_known_at_lookahead = known_at_lookahead.loc[
train_features.index
]
- train_known_at_position = (
- train_positions.to_numpy(dtype=np.int64)
- + train_known_at_lookahead.to_numpy(dtype=np.int64)
- )
+ train_known_at_position = train_positions.to_numpy(
+ dtype=np.int64
+ ) + train_known_at_lookahead.to_numpy(dtype=np.int64)
keep_mask &= train_known_at_position < first_test_position
else:
_log_known_at_none_once(dk.pair, "train_test_split causal guard")
)
if known_at_lookahead is not None:
train_known_at_lookahead = known_at_lookahead.iloc[train_idx]
- train_known_at_position = (
- train_positions.to_numpy(dtype=np.int64)
- + train_known_at_lookahead.to_numpy(dtype=np.int64)
- )
+ train_known_at_position = train_positions.to_numpy(
+ dtype=np.int64
+ ) + train_known_at_lookahead.to_numpy(dtype=np.int64)
keep_mask = train_known_at_position < first_test_position
(
train_features,
selection_method,
keep_fraction,
)
- elif (
- threshold_method in QuickAdapterRegressorV3._SKIMAGE_THRESHOLD_METHODS_SET
- ):
+ elif threshold_method in QuickAdapterRegressorV3._SKIMAGE_THRESHOLD_METHODS_SET:
return QuickAdapterRegressorV3.skimage_min_max(
pred_label,
threshold_method,
drop_mask = np.asarray(drop_mask)
if drop_mask.shape != arr.shape:
raise ValueError(
- f"{context}: drop_mask shape {drop_mask.shape} != arr "
- f"shape {arr.shape}"
+ f"{context}: drop_mask shape {drop_mask.shape} != arr shape {arr.shape}"
)
if not np.issubdtype(drop_mask.dtype, np.bool_):
raise ValueError(
- f"{context}: drop_mask dtype {drop_mask.dtype} is not "
- f"boolean"
+ f"{context}: drop_mask dtype {drop_mask.dtype} is not boolean"
)
safe = np.where(drop_mask, 0.0, safe)
total = safe.sum()
n = base_weights.shape[0]
arr = np.asarray(label_weights, dtype=float)
if arr.shape != (n,):
- raise ValueError(
- f"{context}: label_weights shape {arr.shape}, expected ({n},)"
- )
+ raise ValueError(f"{context}: label_weights shape {arr.shape}, expected ({n},)")
drop_mask = ~np.isfinite(arr) | (arr <= 0.0)
if drop_mask.all():
raise LabelWeightSupportError(