MilesCranmer commited on
Commit
47ad683
1 Parent(s): c2379a1

refactor: NDArray to ndarray for string type

Browse files
Files changed (1) hide show
  1. pysr/utils.py +2 -2
pysr/utils.py CHANGED
@@ -3,11 +3,11 @@ import re
3
  from pathlib import Path
4
  from typing import Any, List, TypeVar, Union
5
 
6
- from numpy.typing import NDArray
7
  from sklearn.utils.validation import _check_feature_names_in # type: ignore
8
 
9
  T = TypeVar("T", bound=Any)
10
- ArrayLike = Union[NDArray[T], List[T]]
11
 
12
 
13
  def _csv_filename_to_pkl_filename(csv_filename: Union[str, Path]) -> Union[str, Path]:
 
3
  from pathlib import Path
4
  from typing import Any, List, TypeVar, Union
5
 
6
+ from numpy import ndarray
7
  from sklearn.utils.validation import _check_feature_names_in # type: ignore
8
 
9
  T = TypeVar("T", bound=Any)
10
+ ArrayLike = Union[ndarray, List[T]]
11
 
12
 
13
  def _csv_filename_to_pkl_filename(csv_filename: Union[str, Path]) -> Union[str, Path]: