Skip to content

Commit 5792a6f

Browse files
committed
Fix template for operators without parameters
1 parent 6eadede commit 5792a6f

2 files changed

Lines changed: 7 additions & 2 deletions

File tree

metafold/func.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,17 +101,18 @@ def __call__(self, eval_: Evaluator) -> TypedResult[Literal[FuncType.FLOAT]]:
101101

102102

103103
class ComputeCurvatures_Parameters(TypedDict, total=False):
104+
pass
104105

105106

106107
class ComputeCurvatures(TypedFunc[Literal[FuncType.VEC3F]]):
107108
def __init__(
108109
self,
109-
samples: Func,
110+
volume: Func,
110111
parameters: Optional[ComputeCurvatures_Parameters] = None,
111112
):
112113
self.inputs: Optional[dict[str, Func]]
113114
self.inputs = {
114-
"Samples": samples,
115+
"Volume": volume,
115116
}
116117
self.assets: Optional[Assets]
117118
self.assets = None

metafold/func.py.in

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ def PointSource(eval_: Evaluator) -> TypedResult[Literal[FuncType.VEC3F]]:
7777

7878

7979
class {{ $type }}_Parameters(TypedDict, total=False):
80+
{{- if len $op.Params }}
8081
{{- range $name, $param := $op.Params }}
8182
{{- $paramType := parameterType $type $name $param }}
8283
{{ $name }}: {{ if $param.IsArray -}}
@@ -85,6 +86,9 @@ Union[list[{{ $paramType }}], "np.ndarray"]
8586
{{ $paramType }}
8687
{{- end }}
8788
{{- end }}
89+
{{- else }}
90+
pass
91+
{{- end }}
8892

8993

9094
class {{ $type }}(

0 commit comments

Comments
 (0)