creation seems to work

molpopgen · Aug 18, 2023 · 9c73507 · 9c73507
1 parent f75b85a
commit 9c73507
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/fwdpy11/genetic_values.py b/fwdpy11/genetic_values.py
@@ -141,13 +141,26 @@ def __eq__(self, other):
 
 @attr.s(auto_attribs=True, frozen=True, repr_ns="fwdpy11")
 class GaussianStabilizingSelection(_ll_GaussianStabilizingSelection):
+    classname: GeneticValueIsTrait
+    optima: list
+
+    def __attrs_post_init__(self):
+        inner = self.classname(self.optima)
+        super(GaussianStabilizingSelection, self).__init__(inner)
+
     @classmethod
     def single_trait(cls, optima):
-        raise NotImplementedError
+        new_optima = []
+        for o in optima:
+            if o.when is None:
+                new_optima.append(Optimum(o.optimum, o.VS, 0))
+            else:
+                new_optima.append(o)
+        return cls(classname=_ll_GSSmo, optima=new_optima)
 
     @classmethod
     def pleiotropy(cls, optima):
-        raise NotImplementedError
+        return cls(classname=_ll_MultivariateGSSmo, optima=optima)
 
 
 @attr_add_asblack