From 6101bf850c377cf1b478dfdb2c0bc13b84a669f0 Mon Sep 17 00:00:00 2001
From: einarr <einarr@pvv.ntnu.no>
Date: Wed, 28 Feb 2007 16:33:11 +0000
Subject: [PATCH] Changed CategoryDataset ftsv files to be written with 0 and 1
instead of True and False.
---
fluents/dataset.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/fluents/dataset.py b/fluents/dataset.py
index b7fde34..de3f566 100644
--- a/fluents/dataset.py
+++ b/fluents/dataset.py
@@ -412,6 +412,9 @@ def write_ftsv(fd, ds):
# Write data
m = ds.asarray()
+ if type == 'category':
+ m = m.astype('i')
+
y, x = m.shape
for j in range(y):
for i in range(x):