diff --git a/pkgs/development/python-modules/sagemaker-mlflow/default.nix b/pkgs/development/python-modules/sagemaker-mlflow/default.nix
index d626f55d6709..8da11f1551e4 100644
--- a/pkgs/development/python-modules/sagemaker-mlflow/default.nix
+++ b/pkgs/development/python-modules/sagemaker-mlflow/default.nix
@@ -17,21 +17,16 @@
 
 buildPythonPackage rec {
   pname = "sagemaker-mlflow";
-  version = "0.1.1";
+  version = ".0.2.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "aws";
     repo = "sagemaker-mlflow";
     tag = "v${version}";
-    hash = "sha256-mHwlP1bVkUiT6RbVf8YLHG+tzkw5+UVrPzcExgcEoJM=";
+    hash = "sha256-EmfEqL+J+cZVdBfUJtAPHpUZCoDV4X1yRfVJYWky1HU=";
   };
 
-  postPatch = ''
-    substituteInPlace VERSION \
-      --replace-fail "${version}.dev0" "${version}"
-  '';
-
   build-system = [
     setuptools
   ];
@@ -41,9 +36,7 @@ buildPythonPackage rec {
     mlflow
   ];
 
-  pythonImportsCheck = [
-    "sagemaker_mlflow"
-  ];
+  pythonImportsCheck = [ "sagemaker_mlflow" ];
 
   nativeCheckInputs = [
     pytestCheckHook