Skip to content

Commit ec7e362

Browse files
kyleabeauchamptwiecki
authored andcommitted
Another fix for text floatX behavior
1 parent 1dc2da0 commit ec7e362

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

pymc3/backends/text.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
from ..backends import base, ndarray
2323
from . import tracetab as ttab
24+
from ..theanof import floatX
2425

2526

2627
class Text(base.BaseTrace):
@@ -104,6 +105,10 @@ def close(self):
104105
def _load_df(self):
105106
if self.df is None:
106107
self.df = pd.read_csv(self.filename)
108+
for key, dtype in self.df.dtypes.items():
109+
if "float" in str(dtype):
110+
self.df[key] = floatX(self.df[key])
111+
107112

108113
def __len__(self):
109114
if self.filename is None:

0 commit comments

Comments
 (0)