test comment removed

poedator · poedator · commit 458642627b3c · 2023-11-10T16:52:01.000+03:00
diff --git a/tests/test_linear4bit.py b/tests/test_linear4bit.py
@@ -30,7 +30,7 @@ def test_linear_serialization(quant_type, compress_statistics, bias):
         compute_dtype=compute_dtype,
         compress_statistics=compress_statistics,
         quant_type=quant_type,
-        device="meta",  # TODO: consider both CPU, meta and CUDA creation
+        device="meta",
     )
     new_weight = bnb.nn.Params4bit(data=linear.weight, requires_grad=False)
     linear_q.weight = new_weight

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ def test_linear_serialization(quant_type, compress_statistics, bias):`
`30`	`30`	`compute_dtype=compute_dtype,`
`31`	`31`	`compress_statistics=compress_statistics,`
`32`	`32`	`quant_type=quant_type,`
`33`		`- device="meta", # TODO: consider both CPU, meta and CUDA creation`
	`33`	`+ device="meta",`
`34`	`34`	`)`
`35`	`35`	`new_weight = bnb.nn.Params4bit(data=linear.weight, requires_grad=False)`
`36`	`36`	`linear_q.weight = new_weight`