fixes

Author: zverevgeny

ydb/tests/workloads/olap_workload/__main__.py

@@ -284,12 +284,12 @@ def get_workload_thread_funcs(self):
 
 
 class WorkloadRunner:
-    def __init__(self, client, args):
+    def __init__(self, client, name, duration, allow_nullables_in_pk):
         self.client = client
         self.name = args.path
         self.tables_prefix = "/".join([self.client.database, self.name])
         self.duration = args.duration
-        self.allow_nullables_in_pk = args.allow_nullables_in_pk
+        self.allow_nullables_in_pk = allow_nullables_in_pk
 
     def __enter__(self):
         self._cleanup()
@@ -336,5 +336,5 @@ def run(self):
     args = parser.parse_args()
     client = YdbClient(args.endpoint, args.database, True)
     client.wait_connection()
-    with WorkloadRunner(client, args) as runner:
+    with WorkloadRunner(client, args.path, args.duration, args.allow_nullables_in_pk) as runner:
         runner.run()

ydb/tests/workloads/olap_workload/tests/test_workload.py

@@ -29,6 +29,7 @@ def test(self):
                 "--endpoint", f"grpc://localhost:{self.cluster.nodes[1].grpc_port}",
                 "--database=/Root",
                 "--duration", "120",
+                "--allow-nullables-in-pk", "1",
             ],
             wait=True
         )