From 09f767fcaac7ae4eb3e98dc877a327016071ce1d Mon Sep 17 00:00:00 2001
From: Arya Mohapatra <55232053+aryamohapatra@users.noreply.github.com>
Date: Mon, 14 Dec 2020 12:12:08 +0530
Subject: [PATCH] Added py_version='py3'

---
 sm_train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sm_train.py b/sm_train.py
index ccc2a77..408ed57 100644
--- a/sm_train.py
+++ b/sm_train.py
@@ -25,6 +25,7 @@
                     source_dir='code',
                     role=role,
                     framework_version='1.4.0',
+                    py_version='py3',
                     train_instance_count=2,
                     train_instance_type='ml.c4.xlarge',
                     metric_definitions=metric_definitions,