Added a case to ensure even supplied types get original_type.data_type

howsoai · Sep 26, 2024 · 4f682e4 · 4f682e4
1 parent 46d8bd0
commit 4f682e4
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/features/sources/Base.ts b/src/features/sources/Base.ts
@@ -39,15 +39,14 @@ export abstract class InferFeatureAttributesBase {
     const columns = await this.getFeatureNames();
 
     const getFeatureAttributes = async (featureName: string): Promise<FeatureAttributes | undefined> => {
-      if (featureName in attributes && attributes[featureName]?.type) {
-        return undefined;
-      }
-
       const originalFeatureType = await this.getOriginalFeatureType(featureName);
       const featureAttributes: Omit<FeatureAttributes, "type"> = {
         original_type: originalFeatureType,
       };
+
       switch (true) {
+        case !!attributes[featureName]?.type:
+          return { ...featureAttributes, type: attributes[featureName]?.type };
         case featureName in ordinalFeatureValues:
           return { ...featureAttributes, type: "ordinal", bounds: { allowed: ordinalFeatureValues[featureName] } };
         case originalFeatureType?.data_type === "numeric":