From 0dec7c0e9e21dabfa433d3bdd936da9eba160b72 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 13 May 2020 20:32:49 +0300
Subject: [PATCH] Add Parquet framework and model type

---
 trains/model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/trains/model.py b/trains/model.py
index b9b294c7..50669e89 100644
--- a/trains/model.py
+++ b/trains/model.py
@@ -47,6 +47,7 @@ class Framework(Options):
     paddlepaddle = 'PaddlePaddle'
     scikitlearn = 'ScikitLearn'
     xgboost = 'XGBoost'
+    parquet = 'Parquet'
 
     __file_extensions_mapping = {
         '.pb': (tensorflow, tensorflowjs, onnx, ),
@@ -74,6 +75,7 @@ class Framework(Options):
         '.cfg': (darknet, ),
         '__model__': (paddlepaddle, ),
         '.pkl': (scikitlearn, keras, xgboost),
+        '.parquet': (parquet),
     }
 
     @classmethod