Merge pull request #91 from dengzq1234/main

allow comparing in datatype None
compgenomicslab · Aug 29, 2024 · 18d8e35 · 18d8e35
2 parents 3f35366 + 46de27b
commit 18d8e35
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/treeprofiler/layouts/staple_layouts.py b/treeprofiler/layouts/staple_layouts.py
@@ -311,7 +311,7 @@ def set_node_style(self, node):
                     tooltip += f'<b>{node.name}</b><br>'
                 if self.heatmap_prop:
                     tooltip += f'<br>{self.heatmap_prop}: {heatmap_num}<br>'
-
+                
                 gradient_color = self.value_color.get(heatmap_num)
 
                 if gradient_color:

diff --git a/treeprofiler/src/utils.py b/treeprofiler/src/utils.py
@@ -93,7 +93,7 @@ def counter_call(node, internal_prop, leaf_prop, datatype, operator_string, righ
 
 def call(node, prop, datatype, operator_string, right_value):
     num_operators = [ '<', '<=', '>', '>=' ] 
-    if datatype == str:
+    if datatype == str or datatype is None:
         if operator_string in num_operators:
             return False
         elif operator_string == 'contains':
@@ -107,7 +107,6 @@ def call(node, prop, datatype, operator_string, right_value):
                 return left_value in right_value 
         else:
             left_value = node.props.get(prop)
-
             if left_value:
                 return operator_dict[operator_string](left_value, right_value)