Fix DirichletElement pickle

bilby/core/prior/conditional.py

@@ -372,6 +372,7 @@ class DirichletElement(ConditionalBeta):
             label + str(ii) for ii in range(order)
         ]
         self.__class__.__name__ = 'DirichletElement'
+        self.__class__.__qualname__ = 'DirichletElement'
 
     def dirichlet_condition(self, reference_parms, **kwargs):
         remaining = 1 - sum(

test/core/prior/conditional_test.py

@@ -5,6 +5,7 @@ from unittest import mock
 
 import numpy as np
 import pandas as pd
+import pickle
 
 import bilby
 
@@ -472,6 +473,10 @@ class TestDirichletPrior(unittest.TestCase):
         test = bilby.core.prior.PriorDict.from_json(filename="priors/test_prior.json")
         self.assertEqual(self.priors, test)
 
+    def test_pickle(self):
+        """Assert can be pickled (needed for use with bilby_pipe)"""
+        pickle.dumps(self.priors)
+
 
 if __name__ == "__main__":
     unittest.main()