relationalai 0.12.0__py3-none-any.whl → 0.12.2__py3-none-any.whl

relationalai/semantics/lqp/model2lqp.py

@@ -102,6 +102,7 @@ def _get_export_reads(export_ids: list[tuple[lqp.RelationId, int, lqp.Type]]) ->
         data_columns=csv_columns,
         compression="gzip",
         partition_size=200,
+        syntax_escapechar='"',  # To follow Snowflake's expected format
         meta=None,
     )
     reads.append(lqp.Read(read_type=lqp.Export(config=export_csv_config, meta=None), meta=None))
@@ -579,8 +580,9 @@ def get_relation_id(ctx: TranslationCtx, relation: ir.Relation, projection: list
     if relation.id in ctx.def_names.id_to_name:
         unique_name = ctx.def_names.id_to_name[relation.id]
     else:
-        prefix = helpers.relation_name_prefix(relation)
-        unique_name = ctx.def_names.get_name_by_id(relation.id, prefix + relation.name)
+        name = helpers.relation_name_prefix(relation) + relation.name
+        name = helpers.sanitize(name)
+        unique_name = ctx.def_names.get_name_by_id(relation.id, name)
 
     return utils.gen_rel_id(ctx, unique_name, types)
 

relationalai/semantics/metamodel/executor.py

@@ -56,5 +56,6 @@ class Executor():
             rich.print(f"[black]{df}[/black]")
         if not df.empty:
             for col in extra_cols:
-                df = df.drop(col, axis=1)
+                if col in df.columns:
+                    df = df.drop(col, axis=1)
         return df

relationalai/semantics/metamodel/rewrite/flatten.py

@@ -558,22 +558,23 @@ class Flatten(Pass):
     def rewrite_wide_output(self, output: ir.Output):
         assert(output.keys)
 
-        # only prefix keys that are not already in the output
-        prefix_keys = []
+        # only append keys that are not already in the output
+        suffix_keys = []
         for key in output.keys:
             if all([val is not key for _, val in output.aliases]):
-                prefix_keys.append(key)
+                suffix_keys.append(key)
 
         aliases: OrderedSet[Tuple[str, ir.Value]] = ordered_set()
-        # add the keys to the output
-        for key in prefix_keys:
-            aliases.add((key.name, key))
 
         # add the remaining args, unless it is already a key
         for name, val in output.aliases:
-            if not isinstance(val, ir.Var) or val not in prefix_keys:
+            if not isinstance(val, ir.Var) or val not in suffix_keys:
                 aliases.add((name, val))
 
+        # add the keys to the output
+        for key in suffix_keys:
+            aliases.add((key.name, key))
+
         # TODO - we are assuming that the Rel compiler will translate nullable lookups
         # properly, returning a `Missing` if necessary, like this:
         # (nested_192(_adult, _adult_name) or (not nested_192(_adult, _) and _adult_name = Missing)) and