fix type of DifferentiableFn

xieyuheng · Jun 6, 2024 · 94df3bf · 94df3bf
1 parent c93c35a
commit 94df3bf
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/system-a/gradient/gradient.ts b/src/system-a/gradient/gradient.ts
@@ -3,7 +3,6 @@ import {
   scalarLink,
   scalarTruncate,
   tensorMap,
-  type Scalar,
   type Tensor,
 } from "../tensor/index.js"
 import {
@@ -14,9 +13,7 @@ import {
 
 // The effect of `gradient` on a `DifferentiableFn`
 // is `sum` of all elements of it's result tensor.
-export type DifferentiableFn =
-  | ((...args: Array<Tensor>) => Tensor)
-  | ((...args: Array<Scalar>) => Tensor)
+export type DifferentiableFn = (...args: Array<Tensor>) => Tensor
 
 export function gradient(fn: DifferentiableFn, args: Array<Tensor>): Tensor {
   const wrt = tensorMap(args, scalarTruncate)