executing readme: 'not supported between instances of 'int' and 'NoneType'

[ush19]

• SDV version: 0.1.1
• Python version: 3.7.3
• Operating System: macOS Mojave 10.14.5

Description

Hi, I'm trying to execute the readme instruction code and I'm seeing the following error when I execute:

samples = data_vault.sample_all()

I don't understand why num_rows is being read as a NoneType even though it should be an int but designated as a string in the comment block.

Traceback

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-4-27b4127b7320> in <module>
----> 1 samples = data_vault.sample_all()

~/Documents/igor_workspace/sdv_validation/SDV/sdv/sdv.py in sample_all(self, num_rows, reset_primary_keys)
     97             raise NotFittedError('SDV instance has not been fitted')
     98 
---> 99         return self.sampler.sample_all(num_rows, reset_primary_keys=reset_primary_keys)
    100 
    101     def save(self, filename):

~/Documents/igor_workspace/sdv_validation/SDV/sdv/sampler.py in sample_all(self, num_rows, reset_primary_keys)
    654         for table in tables:
    655             if not self.dn.get_parents(table):
--> 656                 self.sample_rows(table, num_rows, sampled_data=sampled_data)
    657 
    658         return sampled_data

~/Documents/igor_workspace/sdv_validation/SDV/sdv/sampler.py in sample_rows(self, table_name, num_rows, reset_primary_keys, sample_children, sampled_data)
    606         if sample_children:
    607             for child in self.dn.get_children(table_name):
--> 608                 self.sample_rows(child, sample_children=True, sampled_data=sampled_data)
    609 
    610         return sampled_data

~/Documents/igor_workspace/sdv_validation/SDV/sdv/sampler.py in sample_rows(self, table_name, num_rows, reset_primary_keys, sample_children, sampled_data)
    559             self._reset_primary_keys_generators()
    560 
--> 561         pk_name, pk_values = self._get_primary_keys(table_name, num_rows)
    562         parent_row = self._get_parent_row(table_name)
    563 

~/Documents/igor_workspace/sdv_validation/SDV/sdv/sampler.py in _get_primary_keys(self, table_name, num_rows)
    231                 remaining = self.remaining_primary_key[table_name]
    232 
--> 233             if remaining < num_rows:
    234                 raise ValueError(
    235                     'Not enough unique values for primary key of table {} with regex {}'

TypeError: '<' not supported between instances of 'int' and 'NoneType'