From 2a8f5edea52074d1a4b4cbb169567d300cf18970 Mon Sep 17 00:00:00 2001 From: Allen Goodman Date: Mon, 22 Apr 2024 15:51:56 -0400 Subject: [PATCH] beignet.func.space --- src/beignet/func/_space.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/beignet/func/_space.py b/src/beignet/func/_space.py index a419bc5a97..5491756918 100644 --- a/src/beignet/func/_space.py +++ b/src/beignet/func/_space.py @@ -297,7 +297,7 @@ def shift_fn(a: Tensor, b: Tensor, **_) -> Tensor: return displacement_fn, shift_fn - def _displacement_fn( + def displacement_fn( a: Tensor, b: Tensor, *, @@ -328,4 +328,4 @@ def shift_fn(a: Tensor, b: Tensor, **_) -> Tensor: def shift_fn(a: Tensor, b: Tensor, **_) -> Tensor: return a + b - return _displacement_fn, shift_fn + return displacement_fn, shift_fn