We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f805182 commit c9563f1Copy full SHA for c9563f1
1 file changed
pysr/export.py
@@ -2,6 +2,7 @@
2
3
import copy
4
from collections.abc import Callable
5
+from typing import Any, cast
6
7
import numpy as np
8
import pandas as pd
@@ -82,9 +83,10 @@ def add_export_formats(
82
83
index=output.index,
84
)
85
86
+ exports_any = cast(Any, exports)
87
if output_jax_format:
- exports["jax_format"] = jax_format
88
+ exports_any["jax_format"] = jax_format
89
if output_torch_format:
- exports["torch_format"] = torch_format
90
+ exports_any["torch_format"] = torch_format
91
92
return exports
0 commit comments