From 0616d3e73bfe0baf7525877dbf6acab697395eb9 Mon Sep 17 00:00:00 2001
From: Thiago Franco de Moraes <totonixsame@gmail.com>
Date: Thu, 16 Dec 2021 22:56:32 -0300
Subject: [PATCH] Brain segmentation was not working without pytorch installed

---
 invesalius/gui/brain_seg_dialog.py | 1 +
 1 file changed, 1 insertion(+), 0 deletions(-)

diff --git a/invesalius/gui/brain_seg_dialog.py b/invesalius/gui/brain_seg_dialog.py
index 577c92f..ea725d5 100644
--- a/invesalius/gui/brain_seg_dialog.py
+++ b/invesalius/gui/brain_seg_dialog.py
@@ -21,6 +21,7 @@ from invesalius.gui import dialogs
 HAS_THEANO = bool(importlib.util.find_spec("theano"))
 HAS_PLAIDML = bool(importlib.util.find_spec("plaidml"))
 PLAIDML_DEVICES = {}
+TORCH_DEVICES = {}
 
 try:
     import torch
--
libgit2 0.21.2