Merge pull request #729 from valory-xyz/feat/push-custom-components

Add support for pushing custom components
valory-xyz · Mar 27, 2024 · 1bb984b · 1bb984b
2 parents 47c657e + cb36d3a
commit 1bb984b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/aea/cli/utils/click_utils.py b/aea/cli/utils/click_utils.py
@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 # ------------------------------------------------------------------------------
 #
-#   Copyright 2021-2023 Valory AG
+#   Copyright 2021-2024 Valory AG
 #   Copyright 2018-2020 Fetch.AI Limited
 #
 #   Licensed under the Apache License, Version 2.0 (the "License");
@@ -40,6 +40,7 @@
     CONFIG_FILE_TO_PACKAGE_TYPE,
     CONNECTION,
     CONTRACT,
+    CUSTOM,
     DEFAULT_AEA_CONFIG_FILE,
     PROTOCOL,
     SKILL,
@@ -382,7 +383,7 @@ def wrapper(f: Callable) -> Callable:
 
         f = argument(
             "component_type",
-            type=click.Choice((CONNECTION, CONTRACT, PROTOCOL, SKILL)),
+            type=click.Choice((CONNECTION, CONTRACT, PROTOCOL, CUSTOM, SKILL)),
             required=True,
         )(f)