-
- diff --git a/yt/analysis_modules/halo_finding/rockstar/rockstar.py b/yt/analysis_modules/halo_finding/rockstar/rockstar.py
- index 78ec380fd..de78f6140 100644
- --- a/yt/analysis_modules/halo_finding/rockstar/rockstar.py
- +++ b/yt/analysis_modules/halo_finding/rockstar/rockstar.py
- @@ -214,7 +214,7 @@ class RockstarHaloFinder(ParallelAnalysisInterface):
- ts = DatasetSeries([ts])
- self.ts = ts
- self.particle_type = particle_type
- - self.outbase = outbase
- + self.outbase = six.b(outbase)
- self.min_halo_size = min_halo_size
- if force_res is None:
- tds = ts[-1] # Cache a reference
- @@ -296,7 +296,8 @@ class RockstarHaloFinder(ParallelAnalysisInterface):
- server_address, port = None, None
- self.server_address, self.port = self.comm.mpi_bcast(
- (server_address, port))
- - self.port = str(self.port)
- + self.server_address = six.b(self.server_address)
- + self.port = six.b(str(self.port))
-
- def run(self, block_ratio = 1, callbacks = None, restart = False):
- """
- @@ -328,8 +329,8 @@ class RockstarHaloFinder(ParallelAnalysisInterface):
- else:
- restart_num = 0
- self.handler.setup_rockstar(
- - six.b(self.server_address),
- - six.b(self.port),
- + self.server_address,
- + self.port,
- num_outputs, self.total_particles,
- self.particle_type,
- particle_mass = self.particle_mass,
- @@ -338,7 +339,7 @@ class RockstarHaloFinder(ParallelAnalysisInterface):
- num_writers = self.num_writers,
- writing_port = -1,
- block_ratio = block_ratio,
- - outbase = six.b(self.outbase),
- + outbase = self.outbase,
- force_res = self.force_res,
- callbacks = callbacks,
- restart_num = restart_num,
- @@ -351,7 +352,7 @@ class RockstarHaloFinder(ParallelAnalysisInterface):
- os.makedirs(self.outbase)
- # Make a record of which dataset corresponds to which set of
- # output files because it will be easy to lose this connection.
- - fp = open(self.outbase + '/datasets.txt', 'w')
- + fp = open(self.outbase.decode() + '/datasets.txt', 'w')
- fp.write("# dsname\tindex\n")
- for i, ds in enumerate(self.ts):
- dsloc = path.join(path.relpath(ds.fullpath), ds.basename)
-