Skip to content

Commit 2816b69

Browse files
authored
Removes pandas dependence in data module (#5503)
1 parent ffc3709 commit 2816b69

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

pymc/data.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
import aesara
2626
import aesara.tensor as at
2727
import numpy as np
28-
import pandas as pd
2928

3029
from aesara.compile.sharedvalue import SharedVariable
3130
from aesara.graph.basic import Apply
@@ -472,7 +471,7 @@ def determine_coords(model, value, dims: Optional[Sequence[str]] = None) -> Dict
472471
coords = {}
473472

474473
# If value is a df or a series, we interpret the index as coords:
475-
if isinstance(value, (pd.Series, pd.DataFrame)):
474+
if hasattr(value, "index"):
476475
dim_name = None
477476
if dims is not None:
478477
dim_name = dims[0]
@@ -482,7 +481,7 @@ def determine_coords(model, value, dims: Optional[Sequence[str]] = None) -> Dict
482481
coords[dim_name] = value.index
483482

484483
# If value is a df, we also interpret the columns as coords:
485-
if isinstance(value, pd.DataFrame):
484+
if hasattr(value, "columns"):
486485
dim_name = None
487486
if dims is not None:
488487
dim_name = dims[1]
@@ -501,7 +500,7 @@ def determine_coords(model, value, dims: Optional[Sequence[str]] = None) -> Dict
501500
for size, dim in zip(value.shape, dims):
502501
coord = model.coords.get(dim, None)
503502
if coord is None:
504-
coords[dim] = pd.RangeIndex(size, name=dim)
503+
coords[dim] = range(size)
505504

506505
return coords
507506

0 commit comments

Comments
 (0)