small import fix to makeinifile script

parent 94a8e0ee
......@@ -7,7 +7,7 @@ import os
import sys
import time
import itertools
from numpy import *
import numpy as np
from shutil import copy2, rmtree, move
try:
......@@ -70,7 +70,7 @@ attributes_tags = ['group-pre_movement_mean',
'source-agents_max',
]
import numpy as np
# cor_tags = np.unique([att.split("_")[0] for att in attributes_tags]).astype(str)a
cor_tags = [att.split("-")[0] for att in attributes_tags]
attributes = [att.split(tag+"-")[1] for (att, tag) in zip(attributes_tags, cor_tags)]
......@@ -124,7 +124,7 @@ def get_attribute(node):
value = eval(text)
else:
value = 0
if isinstance(value, list) or isinstance(value, ndarray):
if isinstance(value, list) or isinstance(value, np.ndarray):
if len(value) > 1:
values.append([value, str(node.tag)+"-"+str(node_attrib), node_attrib])
......@@ -148,7 +148,7 @@ def get_product(root):
tag = node.tag
if tag in tags: # ignore tags that are not of interest
d = get_tag(node)
if isinstance(d, list) or isinstance(d, ndarray) or isinstance(d, range):
if isinstance(d, list) or isinstance(d, np.ndarray):
# in case some tags have multiple values
if tag not in input_tags and len(d) > 1:
# ignore lists with one element (equiv to scalars)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment