[GDN] Fix DeprecationWarning

yzhangcs · web-flow · commit 32b1a55bcd1d · 2025-07-15T22:50:07.000+08:00
diff --git a/fla/ops/gated_delta_rule/chunk.py b/fla/ops/gated_delta_rule/chunk.py
@@ -301,7 +301,7 @@ def chunk_gated_delta_rule(
     assert len(beta.shape) == 3, "beta must be of shape [B, T, H] if head_first=False, or [B, H, T] otherwise."
 
     if head_first:
-        raise DeprecationWarning(
+        warnings.warn(
             "head_first is deprecated and will be removed in a future version. "
             "Please use head_first=False for now instead."
         )

Original file line number	Diff line number	Diff line change
`@@ -301,7 +301,7 @@ def chunk_gated_delta_rule(`
`301`	`301`	`assert len(beta.shape) == 3, "beta must be of shape [B, T, H] if head_first=False, or [B, H, T] otherwise."`
`302`	`302`
`303`	`303`	`if head_first:`
`304`		`- raise DeprecationWarning(`
	`304`	`+ warnings.warn(`
`305`	`305`	`"head_first is deprecated and will be removed in a future version. "`
`306`	`306`	`"Please use head_first=False for now instead."`
`307`	`307`	`)`