HE: Enable TX beamformee support with NSTS configuration

Enable TX beamformee support with TX beamformee number of space-time
streams configuration.

Signed-off-by: Jouni Malinen <jouni@codeaurora.org>
diff --git a/sta.c b/sta.c
index f7819dc..84351dc 100644
--- a/sta.c
+++ b/sta.c
@@ -6926,6 +6926,12 @@
 
 	val = get_param(cmd, "BeamformeeSTS");
 	if (val) {
+		if (sta_set_tx_beamformee(dut, intf, 1)) {
+			send_resp(dut, conn, SIGMA_ERROR,
+					"ErrorCode,Failed to set TX beamformee enable");
+			return 0;
+		}
+
 		if (sta_set_beamformee_sts(dut, intf, atoi(val))) {
 			send_resp(dut, conn, SIGMA_ERROR,
 				  "ErrorCode,Failed to set BeamformeeSTS");