Fix XLearner float value

Quantco · Jul 25, 2024 · 8f1e9e0 · 8f1e9e0
1 parent beba0a2
commit 8f1e9e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metalearners/xlearner.py b/metalearners/xlearner.py
@@ -498,7 +498,7 @@ def build_onnx(self, models: Mapping[str, Sequence], output_name: str = "tau"):
             tau_hat_tv = op.add(
                 op.mul(scaled_propensity, tau_hat_control[tv]),
                 op.mul(
-                    op.sub(op.const(1), scaled_propensity),
+                    op.sub(op.constant(value_float=1), scaled_propensity),
                     tau_hat_effect[tv],
                 ),
             )