correct identity_files
diff --git a/reclass_tools/walk_models.py b/reclass_tools/walk_models.py
index 353b0e9..a48c753 100644
--- a/reclass_tools/walk_models.py
+++ b/reclass_tools/walk_models.py
@@ -150,12 +150,12 @@
path = path[:-1]
-def get_all_reclass_params(paths, verbose=False):
+def get_all_reclass_params(paths, identity_files=None, verbose=False):
"""Return dict with all used values for each param"""
#path = '/srv/salt/reclass/classes'
_params = dict()
for path in paths:
- for log in walkfiles(path, verbose=verbose):
+ for log in walkfiles(path, identity_files, verbose):
if log.fname.endswith('.yml'):
model = yaml_read(log.fname)
if model is not None: