Skip to content

Commit c9563f1

Browse files
fix(types): silence pandas-stubs false positive in export
1 parent f805182 commit c9563f1

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

pysr/export.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import copy
44
from collections.abc import Callable
5+
from typing import Any, cast
56

67
import numpy as np
78
import pandas as pd
@@ -82,9 +83,10 @@ def add_export_formats(
8283
index=output.index,
8384
)
8485

86+
exports_any = cast(Any, exports)
8587
if output_jax_format:
86-
exports["jax_format"] = jax_format
88+
exports_any["jax_format"] = jax_format
8789
if output_torch_format:
88-
exports["torch_format"] = torch_format
90+
exports_any["torch_format"] = torch_format
8991

9092
return exports

0 commit comments

Comments
 (0)