Cleanup dispatch

ibis-project · Jul 16, 2018 · db5318e · db5318e
1 parent 97675ea
commit db5318e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ibis/bigquery/datatypes.py b/ibis/bigquery/datatypes.py
@@ -50,8 +50,9 @@ def trans_integer(t, context):
     return 'INT64'
 
 
-@ibis_type_to_bigquery_type.register(dt.UInt64, TypeTranslationContext)
-@ibis_type_to_bigquery_type.register(dt.UInt64, UDFContext)
+@ibis_type_to_bigquery_type.register(
+    dt.UInt64, (TypeTranslationContext, UDFContext)
+)
 def trans_lossy_integer(t, context):
     raise TypeError(
         'Conversion from uint64 to BigQuery integer type (int64) is lossy'