Merge branch 'main' into piyush/models_orch_platform

Author: piyush-zlai

python/src/ai/chronon/repo/__init__.py

@@ -12,12 +12,13 @@
 #     See the License for the specific language governing permissions and
 #     limitations under the License.
 
-from gen_thrift.api.ttypes import ConfType, GroupBy, Join, Model, StagingQuery
+from gen_thrift.api.ttypes import ConfType, GroupBy, Join, Model, ModelTransforms, StagingQuery
 
 JOIN_FOLDER_NAME = "joins"
 GROUP_BY_FOLDER_NAME = "group_bys"
 STAGING_QUERY_FOLDER_NAME = "staging_queries"
 MODEL_FOLDER_NAME = "models"
+MODEL_TRANSFORM_FOLDER_NAME = "model_transforms"
 # TODO - make team part of thrift API?
 TEAMS_FILE_PATH = "teams.json"
 OUTPUT_ROOT = "production"
@@ -29,11 +30,13 @@
     JOIN_FOLDER_NAME: Join,
     STAGING_QUERY_FOLDER_NAME: StagingQuery,
     MODEL_FOLDER_NAME: Model,
+    MODEL_TRANSFORM_FOLDER_NAME: ModelTransforms
 }
 
 FOLDER_NAME_TO_CONF_TYPE = {
     GROUP_BY_FOLDER_NAME: ConfType.GROUP_BY,
     JOIN_FOLDER_NAME: ConfType.JOIN,
     STAGING_QUERY_FOLDER_NAME: ConfType.STAGING_QUERY,
     MODEL_FOLDER_NAME: ConfType.MODEL,
+    MODEL_TRANSFORM_FOLDER_NAME: ConfType.MODEL_TRANSFORMS
 }

python/src/ai/chronon/repo/hub_runner.py

@@ -70,6 +70,8 @@ def get_conf_type(conf):
         return "groupbys"
     elif "compiled/models" in conf:
         return "models"
+    elif "compiled/model_transforms" in conf:
+        return" model_transforms"
     else:
         raise ValueError(f"Unsupported conf type: {conf}")