trim more trailing whitespace..oops

This commit is contained in:
Silver 2025-03-22 16:18:17 +01:00 committed by GitHub
parent 79f460150e
commit 2710f77218
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -126,7 +126,7 @@ class Approximator(nn.Module):
def device(self):
# Get the device of the module (assumes all parameters are on the same device)
return next(self.parameters()).device
def forward(self, x: Tensor) -> Tensor:
x = self.in_proj(x)