From eeda24dbecd5a61dfd102339ceaedee9de9d4526 Mon Sep 17 00:00:00 2001
From: Phillip Chlap <phillip.chlap@unsw.edu.au>
Date: Mon, 4 Dec 2023 12:30:13 +1100
Subject: [PATCH] Add missing import

---
 platipy/imaging/generation/augment.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/platipy/imaging/generation/augment.py b/platipy/imaging/generation/augment.py
index 6a2a8b22..d8fca7ec 100644
--- a/platipy/imaging/generation/augment.py
+++ b/platipy/imaging/generation/augment.py
@@ -16,6 +16,7 @@
 from collections.abc import Iterable
 import random
 import logging
+import sys
 
 from pathlib import Path