fix(ibis_yaml): Distinct: register from_yaml loader (#1303)

fixes #1301

Author: dlovell

python/xorq/common/utils/snowflake_utils.py

@@ -107,9 +107,11 @@ def get_snowflake_last_modification_time(dt):
         .LAST_ALTERED.execute()
         .values
     )
-    if not values:
+    if not len(values):
+        raise ValueError
+    (value, *rest) = values
+    if rest:
         raise ValueError
-    (value,) = values
     return value
 
 

python/xorq/ibis_yaml/translate.py

@@ -1118,6 +1118,13 @@ def _sort_from_yaml(yaml_dict: dict, context: TranslationContext) -> ir.Expr:
     return sort_op.to_expr()
 
 
+@register_from_yaml_handler("Distinct")
+def _distinct_from_yaml(yaml_dict: dict, context: TranslationContext) -> ir.Expr:
+    args = tuple(translate_from_yaml(el, context) for el in yaml_dict["args"])
+    distinct_op = ops.Distinct(*args)
+    return distinct_op.to_expr()
+
+
 @translate_to_yaml.register(ops.SortKey)
 def _sort_key_to_yaml(op: ops.SortKey, context: TranslationContext) -> dict:
     return freeze(