From 9f7606cfba5594157ec51b4c492e8c86868d877a Mon Sep 17 00:00:00 2001
From: Rahul Tuli <rahul@neuralmagic.com>
Date: Fri, 9 Jun 2023 18:15:04 -0400
Subject: [PATCH] Adds back a required argument in default.yaml (#1620)

Updates the hotfix version
---
 src/sparseml/version.py          | 2 +-
 src/sparseml/yolov8/default.yaml | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/sparseml/version.py b/src/sparseml/version.py
index c03409d364b..21ccc54ec43 100644
--- a/src/sparseml/version.py
+++ b/src/sparseml/version.py
@@ -19,7 +19,7 @@
 from datetime import date
 
 
-version_base = "1.5.0"
+version_base = "1.5.1"
 is_release = False  # change to True to set the generated version as a release version
 
 
diff --git a/src/sparseml/yolov8/default.yaml b/src/sparseml/yolov8/default.yaml
index bfcfa9acb8d..edbba465b9b 100644
--- a/src/sparseml/yolov8/default.yaml
+++ b/src/sparseml/yolov8/default.yaml
@@ -2,6 +2,7 @@
 
 recipe: null
 recipe_args: null
+datasets_dir: null
 
 task: "detect" # choices=['detect', 'segment', 'classify', 'init'] # init is a special case. Specify task to run.
 mode: "train" # choices=['train', 'val', 'predict'] # mode to run task in.