diff --git a/src/amplitude/client.py b/src/amplitude/client.py index d028587..a7a8d33 100644 --- a/src/amplitude/client.py +++ b/src/amplitude/client.py @@ -128,7 +128,8 @@ def set_group(self, group_type: str, group_name: Union[str, List[str]], event_op event_options (amplitude.event.EventOptions): Provide additional information for event like user_id. """ - event = IdentifyEvent(groups={group_type: group_name}) + identify = Identify().set(group_type, group_name) + event = IdentifyEvent(groups={group_type: group_name}, user_properties=identify.user_properties) event.load_event_options(event_options) self.track(event) diff --git a/src/test/test_client.py b/src/test/test_client.py index 6301572..1232b20 100644 --- a/src/test/test_client.py +++ b/src/test/test_client.py @@ -194,6 +194,7 @@ def callback_func(event, code, message=None): self.assertEqual("test_user_id", event["user_id"]) self.assertEqual("test_device_id", event["device_id"]) self.assertEqual({"type": ["test_group", "test_group_2"]}, event.groups) + self.assertEqual({"$set": {"type": ["test_group", "test_group_2"]}}, event.user_properties) self.client.configuration.callback = callback_func for use_batch in (True, False):