fix: fixed depth in dgp2wicker (#133)

TRI-ML · Oct 18, 2022 · 2fa0170 · 2fa0170
1 parent fa3aaa2
commit 2fa0170
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