Making it possible to run with only torch

Thiago Franco de Moraes
1 parent cbbe0f41
Showing 1 changed file with 2 additions and 2 deletions Show diff stats
invesalius/gui/frame.py
@@ -790,13 +790,13 @@ class Frame(wx.Frame):
  
     def OnBrainSegmentation(self):
         from invesalius.gui import brain_seg_dialog
-        if brain_seg_dialog.HAS_PLAIDML or brain_seg_dialog.HAS_THEANO:
+        if brain_seg_dialog.HAS_PLAIDML or brain_seg_dialog.HAS_THEANO or brain_seg_dialog.HAS_TORCH:
             dlg = brain_seg_dialog.BrainSegmenterDialog(self)
             dlg.Show()
         else:
             dlg = wx.MessageDialog(self,
                                    _("It's not possible to run brain segmenter because your system doesn't have the following modules installed:") \
-                                   + " PlaidML or Theano" ,
+                                   + " Torch, PlaidML or Theano" ,
                                    "InVesalius 3 - Brain segmenter",
                                    wx.ICON_INFORMATION | wx.OK)
             dlg.ShowModal()
...	...	@@ -790,13 +790,13 @@ class Frame(wx.Frame):
790	790
791	791	def OnBrainSegmentation(self):
792	792	from invesalius.gui import brain_seg_dialog
793		- if brain_seg_dialog.HAS_PLAIDML or brain_seg_dialog.HAS_THEANO:
	793	+ if brain_seg_dialog.HAS_PLAIDML or brain_seg_dialog.HAS_THEANO or brain_seg_dialog.HAS_TORCH:
794	794	dlg = brain_seg_dialog.BrainSegmenterDialog(self)
795	795	dlg.Show()
796	796	else:
797	797	dlg = wx.MessageDialog(self,
798	798	_("It's not possible to run brain segmenter because your system doesn't have the following modules installed:") \
799		- + " PlaidML or Theano" ,
	799	+ + " Torch, PlaidML or Theano" ,
800	800	"InVesalius 3 - Brain segmenter",
801	801	wx.ICON_INFORMATION \| wx.OK)
802	802	dlg.ShowModal()
...	...