From cce1ab66f64007d3b1cd3b8b0251b13e0a5fa912 Mon Sep 17 00:00:00 2001
From: martinRenou <martin.renou@gmail.com>
Date: Thu, 24 Aug 2017 13:37:36 +0100
Subject: [PATCH] Use Type trait class for model_class attributes
---
force_bdss/tests/probe_classes/data_source.py | 4 ++--
force_bdss/tests/probe_classes/kpi_calculator.py | 4 ++--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/force_bdss/tests/probe_classes/data_source.py b/force_bdss/tests/probe_classes/data_source.py
index 8cc11cb..18fccee 100644
--- a/force_bdss/tests/probe_classes/data_source.py
+++ b/force_bdss/tests/probe_classes/data_source.py
@@ -1,4 +1,4 @@
-from traits.api import Bool, Function, Str, Int, on_trait_change
+from traits.api import Bool, Function, Str, Int, on_trait_change, Type
from force_bdss.api import (
BaseDataSourceFactory, BaseDataSourceModel, BaseDataSource,
@@ -47,7 +47,7 @@ class ProbeDataSourceFactory(BaseDataSourceFactory,
id = Str('enthought.test.data_source')
name = Str('test_data_source')
- model_class = ProbeDataSourceModel
+ model_class = Type(ProbeDataSourceModel)
def create_model(self, model_data=None):
return self.model_class(
diff --git a/force_bdss/tests/probe_classes/kpi_calculator.py b/force_bdss/tests/probe_classes/kpi_calculator.py
index c3e4555..951ce9f 100644
--- a/force_bdss/tests/probe_classes/kpi_calculator.py
+++ b/force_bdss/tests/probe_classes/kpi_calculator.py
@@ -1,4 +1,4 @@
-from traits.api import Bool, Function, Str, Int, on_trait_change
+from traits.api import Bool, Function, Str, Int, on_trait_change, Type
from force_bdss.api import (
BaseKPICalculatorFactory, BaseKPICalculatorModel, BaseKPICalculator,
@@ -47,7 +47,7 @@ class ProbeKPICalculatorFactory(BaseKPICalculatorFactory,
id = Str('enthought.test.kpi_calculator')
name = Str('test_kpi_calculator')
- model_class = ProbeKPICalculatorModel
+ model_class = Type(ProbeKPICalculatorModel)
def create_model(self, model_data=None):
return self.model_class(
--
GitLab