From 2dc78933903baa9acc200a40f1795be320d1c297 Mon Sep 17 00:00:00 2001
From: Arun Isaac
Date: Thu, 17 Jul 2025 19:23:30 +0100
Subject: Only output key optionally.

* pyhegp/pyhegp.py (encrypt): Only output key to file optionally.
---
 pyhegp/pyhegp.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyhegp/pyhegp.py b/pyhegp/pyhegp.py
index 2177881..aa4b3fe 100644
--- a/pyhegp/pyhegp.py
+++ b/pyhegp/pyhegp.py
@@ -94,8 +94,7 @@ def pool(pooled_summary_file, summary_files):
               help="Summary statistics file",
               required=True)
 @click.option("--key", "-k", "key_file", type=click.File("w"),
-              help="Output key",
-              required=True)
+              help="Output key")
 @click.option("--output", "-o", "ciphertext_file", type=click.File("w"),
               default="-",
               help="Output ciphertext")
@@ -106,7 +105,8 @@ def encrypt(genotype_file, summary_file, key_file, ciphertext_file):
     key = random_key(rng, len(genotype))
     encrypted_genotype = hegp_encrypt(genotype, summary.mean,
                                       summary.std, key)
-    np.savetxt(key_file, key, delimiter=",", fmt="%f")
+    if key_file:
+        np.savetxt(key_file, key, delimiter=",", fmt="%f")
     np.savetxt(ciphertext_file, encrypted_genotype, delimiter=",", fmt="%f")
 
 if __name__ == "__main__":
-- 
cgit v1.2.3