fix

include/substrait-mlir/Dialect/Substrait/IR/SubstraitTypes.td

@@ -33,6 +33,28 @@ def Substrait_BinaryType : Substrait_Type<"Binary", "binary"> {
   }];
 }
 
+def Substrait_DateType : Substrait_Type<"Date", "date"> {
+  let summary = "Substrait date type";
+  let description = [{
+    This type represents a substrait date type. 
+  }];
+}
+
+def Substrait_DateAttr : Substrait_Attr<"Date", "date",
+  [TypedAttrInterface]> {
+  let summary = "Substrait date type";
+  let description = [{
+    This type represents a substrait date attribute type.
+  }];  
+  let parameters = (ins "int32_t":$value);
+  let assemblyFormat = [{ `<` $value `>` }];
+  let extraClassDeclaration = [{ 
+    ::mlir::Type getType() const {
+      return DateType::get(getContext());
+    }
+  }];
+}
+
 def Substrait_StringType : Substrait_Type<"String", "string"> {
   let summary = "Substrait string type";
   let description = [{
@@ -84,23 +106,6 @@ def Substrait_TimestampTzAttr : Substrait_Attr<"TimestampTz", "timestamp_tz",
   }];
 }
 
-def Substrait_DateType : Substrait_Type<"Date", "date"> {
-  let summary = "Substrait date type";
-  let description = [{
-    This type represents a substrait date type. 
-  }];
-}
-
-def Substrait_DateAttr : Substrait_Attr<"Date", "date",
-  [DeclareAttrInterfaceMethods<TypedAttrInterface>]> {
-  let summary = "Substrait date type";
-  let description = [{
-    This type represents a substrait date attribute type.
-  }];  
-  let parameters = (ins "int32_t":$value);
-  let assemblyFormat = [{ `<` $value `>` }];
-}
-
 /// Currently supported atomic types. These correspond directly to the types in
 /// https://github.com/substrait-io/substrait/blob/main/proto/substrait/type.proto.
 // TODO(ingomueller): Add the other low-hanging fruits here.