fix: fixed depth in dgp2wicker

TRI-ML · Oct 17, 2022 · 4d6375b · 4d6375b
1 parent fa3aaa2
commit 4d6375b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dgp/contribs/dgp2wicker/dgp2wicker/dataset.py b/dgp/contribs/dgp2wicker/dgp2wicker/dataset.py
@@ -82,6 +82,8 @@ def compute_columns(
     if with_ontology_table and requested_annotations is not None:
         for ann in requested_annotations:
             if ann in ANNOTATION_REGISTRY:
+                if ann == 'depth': # DenseDepth does not require an ontology
+                    continue
                 columns_to_load.append(gen_wicker_key('ontology', ann))
 
     return columns_to_load