From f295617bed419770ebb58cb2741d666d7e196028 Mon Sep 17 00:00:00 2001
From: ivanzvonkov <ivan.zvonkov@gmail.com>
Date: Wed, 26 Oct 2022 09:16:37 -0400
Subject: [PATCH] update torchserve

---
 openmapflow/Dockerfile   | 2 +-
 openmapflow/constants.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/openmapflow/Dockerfile b/openmapflow/Dockerfile
index 792deae4..335d3634 100644
--- a/openmapflow/Dockerfile
+++ b/openmapflow/Dockerfile
@@ -1,5 +1,5 @@
 # syntax = docker/dockerfile:experimental
-FROM pytorch/torchserve:0.5.3-cpu
+FROM pytorch/torchserve:0.6.0-cpu
 USER root
 RUN pip3 install --upgrade pip
 ARG VERSION
diff --git a/openmapflow/constants.py b/openmapflow/constants.py
index 6b371398..ee4d0ca4 100644
--- a/openmapflow/constants.py
+++ b/openmapflow/constants.py
@@ -13,7 +13,7 @@
 TEMPLATE_README = TEMPLATES_DIR / "README.md"
 TEMPLATE_DEPLOY_YML = TEMPLATES_DIR / "github-deploy.yaml"
 TEMPLATE_TEST_YML = TEMPLATES_DIR / "github-test.yaml"
-VERSION = "0.2.1rc2"
+VERSION = "0.2.1"
 
 # -------------- Dataframe column names --------------------------------------
 SOURCE = "source"